Cod sursa(job #2168872)

Utilizator andreiutu111Noroc Andrei Mihail andreiutu111 Data 14 martie 2018 12:38:00
Problema Arbori indexati binar Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.6 kb
#include<bits/stdc++.h>
using namespace std;
int N,M,tip,idx,val,Arb[400100];
long long sum;
void Update(int nod,int st,int dr){
    if(st==dr){
        Arb[nod]=val;
        return;
    }
    int mij=(st+dr)/2;
    if(idx<=mij)Update(2*nod,st,mij);
    else Update(2*nod+1,mij+1,dr);
    Arb[nod]=Arb[2*nod]+Arb[2*nod+1];
}
void Update1(int nod,int st,int dr){
    if(st==dr&&st==idx){
        Arb[nod]+=val;
        return;
    }
    int mij=(st+dr)/2;
    if(idx<=mij)Update1(2*nod,st,mij);
    else Update1(2*nod+1,mij+1,dr);
    Arb[nod]=Arb[2*nod]+Arb[2*nod+1];
}
void Query(int nod,int st,int dr){
    if(idx<=st&&dr<=val){
        sum+=Arb[nod];
        return;
    }
    int mij=(st+dr)/2;
    if(idx<=mij)Query(2*nod,st,mij);
    else if(mij<val)Query(2*nod+1,mij+1,dr);
}
int main()
{
    freopen("aib.in","r",stdin);
    freopen("aib.out","w",stdout);
    scanf("%d%d",&N,&M);
    for(int i=1;i<=N;++i){
        scanf("%d",&val);
        idx=i;
        Update(1,1,N);
    }
    while(M--){
        scanf("%d",&tip);
        if(tip==0){
            scanf("%d%d",&idx,&val);
            Update1(1,1,N);
        }else if(tip==1){
            scanf("%d%d",&idx,&val);
            sum=0;
            Query(1,1,N);
            printf("%d\n",sum);
        }else{
            int a;
            scanf("%d",&a);
            idx=1;
            bool ok=1;
            for(int i=1;i<=N&&ok;++i){
                val=i,sum=0,Query(1,1,N);
                if(sum==a)printf("%d\n",i),ok=0;
            }
            if(ok)printf("-1\n");
        }
    }
    return 0;
}