Cod sursa(job #848404)

Utilizator Al3ks1002Alex Cociorva Al3ks1002 Data 5 ianuarie 2013 14:00:53
Problema Arbori indexati binar Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.14 kb
#include<cstdio>
using namespace std;
int n,m,i,cod,a,b,k,v[(1<<17)+5],x,val;
void add(int poz, int val)
{
    for(;poz<=k;poz+=poz&(-poz))
    v[poz]+=val;
}
int sum(int poz)
{
    int ret=0;
    for(;poz;poz-=poz&(-poz)) ret+=v[poz];
    return ret;
}
int query(int val)
{
    if(v[k]<val) return -1;
    int step=k,y=0,poz=0;
    for(;step;step/=2)
        if(y+v[poz+step]<val)
        {
            y+=v[poz+step];
            poz+=step;
        }
    if(sum(poz+1)==val) return poz+1;
    return -1;
}
int main()
{
    freopen("aib.in","r",stdin);
    freopen("aib.out","w",stdout);
    scanf("%d%d",&n,&m);
    k=1<<17;
    while(k/2>=n) k/=2;
    for(i=1;i<=n;i++)
    {
        scanf("%d",&x);
        add(i,x);
    }
    for(;m;m--)
    {
        scanf("%d",&cod);
        if(cod==0)
        {
            scanf("%d%d",&a,&b);
            add(a,b);
            continue;
        }
        if(cod==1)
        {
            scanf("%d%d",&a,&b);
            printf("%d\n",sum(b)-sum(a-1));
            continue;
        }
        scanf("%d",&val);
        printf("%d\n",query(val));
    }
    return 0;
}