Cod sursa(job #1204161)

Utilizator YoChinezuWeng Mihai Alexandru YoChinezu Data 2 iulie 2014 10:43:40
Problema Arbori indexati binar Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.16 kb
#include <cstdio>

using namespace std;

int aib[100005];
int n;

int querry1(int poz){
    int s=0;
    for(;poz>=1;poz-=poz&-poz){
        s+=aib[poz];
    }
    return s;
}

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

int bin(int val){
    int st,dr,last=-1,med;
    st=1;
    dr=n;
    while(st<=dr){
        med=st+(dr-st)/2;
        if(querry1(med)==val){   last=med;
            dr=med-1;
        }
        else
            if(querry1(med)<val)    st=med+1;
        else
            dr=med-1;
    }
    return last;
}

int main(){
    freopen("aib.in", "r", stdin);
    freopen("aib.out", "w", stdout);
    int m,x,a,b;
    
    scanf("%d%d",&n,&m);
    
    for(int i=1;i<=n;++i){
        scanf("%d",&x);
        upgrade(i,x);
    }
    
    for(int i=1;i<=m;++i){
        scanf("%d",&x);
        if(x==0){
            scanf("%d%d",&a,&b);
            upgrade(a,b);
        }else
            if(x==1){
                scanf("%d%d",&a,&b);
                printf("%d\n",querry1(b)-querry1(a-1));
            }else{
                scanf("%d",&a);
                printf("%d\n",bin(a));
            }
    }
    
    return 0;
}