Cod sursa(job #1043848)

Utilizator horatiu11Ilie Ovidiu Horatiu horatiu11 Data 28 noiembrie 2013 23:20:36
Problema Arbori indexati binar Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.2 kb
# include <cstdio>
# define LSB(x) ((-x) & x)
using namespace std;
int n,m,aib[100001],x,op,a,b;
void update(int val, int poz)
{
    int i;
    for(i=poz;i<=n;i+=LSB(i))
        aib[i]+=val;
}
int sum(int poz)
{
    int s=0,i;
    for(i=poz;i>0;i-=LSB(i))
        s+=aib[i];
    return s;
}
int cb(int val)
{
    int i=1,j=n,m,x;
    while(i<= j)
    {
        m=(i+j)>>1;
        x=sum(m);
        if(x==val) return m;
        else if(val<x) j=m-1;
                       else i=m+1;
    }
    return -1;
}
int main()
{
    int i;
    freopen("aib.in","r",stdin);
    freopen("aib.out","w",stdout);
    scanf("%d%d",&n,&m);
    for(i=1;i<=n;++i)
    {
        scanf("%d",&x);
        update(x,i);
    }
    for(i=1;i<=m;++i)
    {
        scanf("%d",&op);
        switch(op)
        {
            case 0:
                scanf("%d%d",&a,&b);
                update(b,a);
                break;
            case 1:
                scanf("%d%d",&a,&b);
                printf("%d\n",sum(b)-sum(a-1));
                break;
            case 2:
                scanf("%d",&a);
                printf("%d\n",cb(a));
                break;
        }
    }
    return 0;
}