Cod sursa(job #2999296)

Utilizator alexcmeciu1Cmeciu Alexandru Cristian alexcmeciu1 Data 10 martie 2023 19:54:56
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.42 kb
#include <bits/stdc++.h>

using namespace std;
ifstream fin("datorii.in");
ofstream fout("datorii.out");
const int nr = 100005;
int a[nr];
int sg[4 * nr];
int n,m;
void citire(){
    fin>>n>>m;
    for(int i=1;i<=n;i++)
        fin>>a[i];
}
void build(int nod, int st, int dr)
{
    if(st==dr)sg[nod]=a[st];
    else
    {
        int mij=(st+dr)/2;
        build(2*nod, st, mij);
        build(2*nod+1, mij+1, dr);
        sg[nod]=sg[nod*2]+sg[nod*2+1];
    }
}
void update(int nod, int st, int dr, int pos, int nv)
{
    if(st==dr)sg[nod]-=nv;
    else
    {
        int mij=(st+dr)/2;
        if(pos<=mij)update(nod*2, st, mij, pos, nv);
        else update(2*nod+1, mij+1, dr, pos, nv);

        sg[nod]=sg[nod*2]+sg[2*nod+1];
    }
}
int query(int nod, int st, int dr, int qst, int qdr)
{
    if(qst<=st && dr<=qdr)return sg[nod];
    else
    {
        int mij=(st+dr)/2;
        if(qdr<=mij)
            return query(nod*2, st, mij, qst, qdr);
        if(mij+1<=qst)
            return query(nod*2+1, mij+1, dr, qst, qdr);

        return query(2*nod, st, mij, qst, qdr)+query(nod*2+1, mij+1, dr, qst, qdr);
    }
}
int main()
{
    citire();
    build(1, 1, n);
    for(int i=1;i<=m;i++){
        int x,y,c;
        fin>>c>>x>>y;
        if(c==1){
            fout<<query(1,1,n,x,y)<<endl;
        }
        else{
            update(1,1,n,x,y);
        }
    }
    return 0;

}