Cod sursa(job #3004764)

Utilizator MihaiCostacheCostache Mihai MihaiCostache Data 16 martie 2023 16:22:57
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.41 kb
#include <bits/stdc++.h>

using namespace std;

ifstream fin("datorii.in");
ofstream fout("datorii.out");
int n, m, v[15001], op, a, b, sol, aint[60001];

void build(int nod, int st, int dr)
{
    if(st==dr)
    {
        aint[nod]=v[st];
    }
    else
    {
        int mid=(st+dr)/2;
        build(2*nod, st, mid);
        build(2*nod+1, mid+1, dr);
        aint[nod]=aint[2*nod]+aint[2*nod+1];
    }
}

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

int query(int nod, int st, int dr, int a, int b)
{
    if(a<=st && dr<=b)
        return aint[nod];
    else
    {
        int mid=(st+dr)/2;
        if(b<=mid)
            return query(2*nod, st, mid, a, b);
        if(a>=mid+1)
            return query(2*nod+1, mid+1, dr, a, b);
        return query(2*nod, st, mid, a, b)+query(2*nod+1, mid+1, dr, a, b);
    }
}

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