Cod sursa(job #2019600)

Utilizator darisavuSavu Daria darisavu Data 8 septembrie 2017 09:32:16
Problema Arbori indexati binar Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.99 kb
#include <fstream>
#define ub(x) (x&(-x))
using namespace std;
ifstream f("aib.in");
ofstream g("aib.out");
int n,aib[100005],k,z,v,t,y;
void update(int poz,int val)
{
    for(int i=poz;i<=n;i=i+ub(i))
    {
        aib[i]=aib[i]+val;
    }
}
int sum(int poz)
{
    int s=0;
    for(int i=poz;i>0;i=i-ub(i))
    {
        s=s+aib[i];
    }
    return s;
}
int doi(int nr)
{
    int st,dr,mij;
    st=1;
    dr=n;
    while(st<=dr)
        {
            mij=(st+dr)/2;
            if(sum(mij)==nr) return mij;
            else if(sum(mij)<nr) st=mij+1;
            else dr=mij-1;
        }
        return -1;

}
int main()
{
    int i;
    f>>n>>k;
    for(i=1;i<=n;i++)
    {
        f>>y;
        update(i,y);
    }
    for(i=1;i<=k;i++)
    {
        f>>z;
        if(z==0) {f>>v>>t;update(v,t);}
        else if(z==1) {f>>v>>t;g<<sum(t)-sum(v-1)<<'\n';}
        else
        {
            f>>v;
            g<<doi(v)<<'\n';
        }
    }
    return 0;
}