Cod sursa(job #1260005)

Utilizator zikade9Irimia Rares zikade9 Data 10 noiembrie 2014 19:55:30
Problema Arbori indexati binar Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.25 kb
#include<cstdio>
#define lsb(x) (x&(-x))
int mij,u,p,j,k,tip,n,m,i, o, a,b,x[100009];
void update(int poz,int val)
{
    for(int i=poz;i<=n;i+=lsb(i))
        x[i]+=val;
}
int query(int poz)
{
    int s=0;
    for(int i=poz;i>=1;i-=lsb(i))
        s+=x[i];
    return s;
}
int main()
{
    freopen("aib.in","r",stdin);
    freopen("aib.out","w",stdout);
    scanf("%d%d",&n,&m);
    for(i=1;i<=n;i++)
    {
        scanf("%d",&a);
        update(i,a);
    }
    for(i=1;i<=m;i++)
    {
        scanf("%d%d",&tip,&a);
        if(tip==0)
        {
            scanf("%d", &b);
            update(a,b);
        }
        else if(tip==1)
        {
            scanf("%d", &b);
            printf("%d\n",query(b)-query(a-1));
        }
        else
        {
            o=0;
            p=1;
            u=n;
            mij=(p+u)/2;
            while(p<=u)
            {
                if(query(mij)>a) u=mij-1;
                else if(query(mij)<a) p=mij+1;
                else
                {
                    o=1;
                    printf("%d\n",mij);
                    break;
                }
                mij=(p+u)/2;
            }
            if(o==0) printf("-1\n");
        }
    }
    return 0;
}