Cod sursa(job #2532129)

Utilizator severutBogdan Sever-Cristian severut Data 27 ianuarie 2020 13:41:11
Problema Datorii Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.27 kb
#include <iostream>
#include <fstream>

using namespace std;

ifstream in("datorii.in");
ofstream out("datorii.out");

int n,m;
int s[15005],arb[100005];
void update(int nod,int st,int dr,int a,int b,int val)
{
    if(a<=st&&b>=dr){
        arb[nod]=s[a]-s[b-1]-val;
        return;
    }
    else{
        int mij=(st+dr)/2;
        if(mij>=a){
            update(2*nod,st,mij,a,b,val);
        }
        if(mij<b){
            update(2*nod+1,mij+1,dr,a,b,val);
        }
        arb[nod]=arb[2*nod]+arb[2*nod+1];
    }
}
void query(int nod,int st,int dr,int a,int b,int &suma)
{
    if(st>=a&&dr<=b){
        suma+=arb[nod];
        return;
    }
    else{
        int mij=(st+dr)/2;
        if(mij>=a){
            query(2*nod,st,mij,a,b,suma);
        }
        if(mij<b){
            query(2*nod+1,mij+1,dr,a,b,suma);
        }

    }
}
int main()
{
    int x,y,cer,suma;
    in>>n>>m;
    for(int i=1;i<=n;i++){
        in>>x;
        s[i]=s[i-1]+x;
        update(1,1,n,i,i,0);
    }
    for(int i=1;i<=m;++i){
        in>>cer>>x>>y;
        if(cer==1){
            suma=0;
            query(1,1,n,x,y,suma);
            out<<suma<<'\n';
        }
        else{
            update(1,1,n,x,x,y);
        }
    }
    return 0;
}