Cod sursa(job #877916)

Utilizator gabriel93Robu Gabriel gabriel93 Data 13 februarie 2013 14:01:43
Problema Arbori indexati binar Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.9 kb
#include<stdio.h>
#define Nmax 200002
using namespace std;
int n,m;
int poz,val,qs,qf,maxim,k,t;
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)
{
    int fs,fd,k1;
    k1=k;
    if(k==1 && arb[nod]==a)
        return c+k;
    if(k==1 && arb[nod]!=a)
        return -1;
    fs=nod<<1;
    fd=fs+1;
    if(arb[fs]<a)
    {
        k=k1>>1;
        search(fd,a-arb[fs],c+k1-k);
    }
    else
    {
        k=k-(k>>1);
        search(fs,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;
            b=search(1,a,0);
            if(b==0)
                b=-1;
            printf("%d\n",b);
            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;
}