Cod sursa(job #2130702)

Utilizator andreiutu111Noroc Andrei Mihail andreiutu111 Data 13 februarie 2018 20:46:09
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.03 kb
#include<bits/stdc++.h>
using namespace std;
int N,M,x,ind,val,s,Arb[4*15001+101],tip;
void Update(int nod,int st,int dr,int var){
    if(st==dr){
        if(var==1)Arb[nod]=val;
        else Arb[nod]-=val;
        return;
    }
    int mij=(st+dr)/2;
    if(ind<=mij)Update(2*nod,st,mij,var);
    else Update(2*nod+1,mij+1,dr,var);
    Arb[nod]=Arb[2*nod]+Arb[2*nod+1];
}
void Query(int nod,int st,int dr){
    if(ind<=st&&dr<=val){
        s+=Arb[nod];
        return;
    }
    int mij=(st+dr)/2;
    if(ind<=mij)Query(2*nod,st,mij);
    if(mij<val)Query(2*nod+1,mij+1,dr);
}
int main()
{
    freopen("datorii.in","r",stdin);
    freopen("datorii.out","w",stdout);
    scanf("%d%d",&N,&M);
    for(int i=1;i<=N;++i){
        scanf("%d",&x);
        ind=i,val=x;
        Update(1,1,N,1);
    }
    while(M--){
        scanf("%d%d%d",&tip,&ind,&val);
        if(tip==0)Update(1,1,N,2);
        else{
            s=0;
            Query(1,1,N);
            printf("%d\n",s);
        }
    }
    return 0;
}