Cod sursa(job #371735)

Utilizator hasegandaniHasegan Daniel hasegandani Data 6 decembrie 2009 17:43:22
Problema Arbori indexati binar Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.35 kb
#include<stdio.h>

#define Nmax 100001

int aib[Nmax],n,m;

void update(int poz,int val)
{
    for(;poz<=n; poz += (poz & -poz))
        aib[poz]+=val;
}

int read(int poz)
{
    int sum=0;
    for( ; poz ; poz -= poz & -poz )
        sum+=aib[poz];
        
    return sum;
}

int find(int val)
{
    int i,step;
    for(step=1; step<n ; step<<=1);
    
    for(i=0; step ; step>>=1)
        if (step+i<=n && aib[i+step]<=val)
            {
            i+=step;
            val-=aib[i];
            if (!val)
                return i;
            }
    return -1;
}

void cit();
void solve();

int main()
{
    cit();
    solve();
    return 0;
}

void cit()
{
    int a;
    freopen("aib.in","r",stdin);
    freopen("aib.out","w",stdout);
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;++i)
        {
        scanf("%d",&a);
        update(i,a);
        }
}

void solve()
{
    int a,b,c;
    for(int i=1;i<=m;++i)
        {
        scanf("%d",&a);
        if (a==0)
            {
            scanf("%d%d",&b,&c);
            update(b,c);
            }
        if (a==1)
            {
            scanf("%d%d",&b,&c);
            printf("%d\n",read(c)-read(b-1));
            }
        if (a==2)
            {
            scanf("%d",&b);
            printf("%d\n",find(b));
            }
        }
}