Cod sursa(job #877874)

Utilizator gabriel93Robu Gabriel gabriel93 Data 13 februarie 2013 13:13:41
Problema Arbori indexati binar Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.79 kb
#include<stdio.h>
#define Nmax 100002
using namespace std;
int n,m;
int poz,val,qs,qf,maxim,k;
int arb[Nmax];

void update(int s,int f,int nod)
{
    int mid;
    if(s==f)
    {
        arb[nod]+=val;
        return;
    }
    mid=s+(f-s)/2;
    if(poz<=mid)
        update(s,mid,nod<<1);
    else
        update(mid+1,f,(nod<<1)+1);
    arb[nod]=arb[nod<<1]+arb[(nod<<1)+1];
}

void query(int s,int f,int nod)
{
    int mid;
    if(qs<=s && f<=qf)
    {
        maxim+=arb[nod];
        return;
    }
    mid=s+(f-s)/2;
    if(qs<=mid)
        query(s,mid,nod<<1);
    if(mid<qf)
        query(mid+1,f,(nod<<1)+1);
}

int search(int nod,int a,int c)
{
    if(a==0)
        return c;
    if(arb[nod]<a)
    {
        a-=arb[nod];
        c+=k;
        search(nod+1,a,c);
    }
    else
    {
        k=k>>1;
        if((nod<<1)>=2*n)
            return c+1;
        else
            search(nod<<1,a,c);
    }

}

void rezolv()
{
    int i,x,a,b;
    scanf("%d %d",&n,&m);
    for(i=1;i<=n;++i)
    {
        scanf("%d ",&x);
        val=x;
        poz=i;
        update(1,n,1);
    }
    for(i=1;i<=m;++i)
    {
        scanf("%d",&x);
        if(x==2)
        {
            scanf("%d",&a);
            k=n;
            printf("%d\n",search(1,a,0));
            continue;
        }
        scanf("%d %d",&a,&b);
        if(x==0)
        {
            poz=a;
            val=b;
            update(1,n,1);
        }
        else
        {
            qs=a;
            qf=b;
            maxim=0;
            query(1,n,1);
            printf("%d\n",maxim);
        }
    }
}



int main()
{
    freopen("aib.in","r",stdin);
   //freopen("aib.out","w",stdout);
    rezolv();
    fclose(stdin);
    fclose(stdout);
    return 0;
}