Cod sursa(job #2316734)

Utilizator PopeangaMihneaPopeanga Mihnea- Stefan PopeangaMihnea Data 12 ianuarie 2019 12:35:10
Problema Datorii Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.1 kb
#include <bits/stdc++.h>

using namespace std;

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

int n, m, x, op, a, b, sol;
int Aint[30001];

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

void querry(int nod, int st, int dr, int qs, int qd)
{
    if(qs<=st && dr<=qd)
    {
        sol=sol+Aint[nod];
        return;
    }
    int mid=(st+dr)/2;
    if(qs<=mid) querry(2*nod, st, mid, qs, qd);
    if(mid<qd) querry(2*nod+1, mid+1, dr, qs, qd);
}

int main()
{
    fin>>n>>m;
    for(int i=1; i<=n; ++i)
    {
        fin>>x;
        update(1, 1, n, i, x);
    }
    for(int i=1; i<=m; ++i)
    {
        fin>>op>>a>>b;
        if(op==0) update(1, 1, n, a, -b);
        if(op==1)
        {
            sol=0;
            querry(1, 1, n, a, b);
            fout<<sol<<"\n";
        }
    }
    return 0;
}