Cod sursa(job #1274406)

Utilizator RazecBotez Cezar Razec Data 23 noiembrie 2014 19:39:07
Problema Arbori indexati binar Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.12 kb
#include<cstdio>
using namespace std;
int n,m,lim,i,val,poz,x,y,aib[100005];
void upd(int poz,int val)
{
    for(int i=poz;i<=n;i+=i&(-i))
        aib[i]+=val;
}
int sum(int poz)
{
    int r=0;
    for(int i=poz;i;i-=i&(-i))
        r+=aib[i];
    return r;
}
int query(int x)
{
    int step,r=0,y=x;
    for(step=lim;step;step>>=1)
        if(r+step<=n && aib[r+step]<x)
        {
            r+=step;
            x-=aib[r];
        }
    if(sum(r+1)==y) return r+1;
    return -1;
}
int main()
{
    freopen("aib.in","r",stdin);
    freopen("aib.out","w",stdout);
    scanf("%d%d",&n,&m);
    for(lim=1;lim<=n;lim<<=1);
    for(i=1;i<=n;i++)
    {
        scanf("%d",&x);
        upd(i,x);
    }
    for(;m;m--)
    {
        scanf("%d",&i);
        if(i==0)
        {
            scanf("%d%d",&poz,&val);
            upd(poz,val);
        }
        else if(i==1)
        {
            scanf("%d%d",&x,&y);
            printf("%d\n",sum(y)-sum(x-1));
        }
        else
        {
            scanf("%d",&x);
            printf("%d\n",query(x));
        }
    }
    return 0;
}