Cod sursa(job #2541921)

Utilizator xXoctavianXxStanescu Matei Octavian xXoctavianXx Data 9 februarie 2020 09:55:54
Problema Datorii Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.24 kb
#include <bits/stdc++.h>

#define nmax 100002

using namespace std;

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

int arb[nmax*4],v[nmax+1];
int n,m,a,b,t;

void update(int st, int dr, int nod, int pos)
{
    if(st==dr)
    {
        arb[nod]=v[pos];
        return;
    }
    int mij=(st+dr)/2;
    if(pos<=mij) update(st,mij,nod*2,pos);
    else update(mij+1,dr,nod*2+1,pos);
    arb[nod] = arb[2*nod]+arb[2*nod+1];
}

int query(int st, int dr, int p1, int p2, int nod)
{
    if(p1==st && p2==dr) return arb[nod];
    else
    {
        int mij=(st+dr)/2;
        if(p2<=mij) return query(st,mij,p1,p2,2*nod);
        else if(p1>mij) return query(mij+1,dr,p1,p2,2*nod+1);
        else
        {
            return query(st,mij,p1,mij,2*nod)+query(mij+1,dr,mij+1,p2,2*nod+1);
        }
    }
}

int main()
{
    fin>>n>>m;
    for(int i=1; i<=n; i++)
    {
        fin>>v[i];
    }
    for(int i=1; i<=n; i++)
    {
        update(1,n,1,i);
    }
    for(int i=0; i<m; i++)
    {
        fin>>t>>a>>b;
        if(t==0)
        {
            fout<<query(1,n,a,b,1)<<"\n";
        }
        if(t==1)
        {
            v[a]-=b;
            update(1,n,1,a);
        }
    }
    return 0;
}