Cod sursa(job #3122150)

Utilizator unomMirel Costel unom Data 17 aprilie 2023 13:57:04
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.77 kb
#include <fstream>

using namespace std;

ifstream in("datorii.in");
ofstream out("datorii.out");
int n, m;
int v[15005], aint[60005];

void build(int node, int left, int right)
{
    if(left == right)
    {
        aint[node] = v[left];
    }
    else
    {
        int m = (left + right) / 2;

        build(2*node, left, m);
        build(2*node+1, m+1, right);

        aint[node] = aint[2*node] + aint[2*node+1];
    }
}

void update(int node, int left, int right, int poz, int val)
{
    if(left == right)
    {
        aint[node] -= val;
    }
    else
    {
        int m = (left + right) / 2;

        if(poz <= m)
        {
            update(2*node, left, m, poz, val);
        }
        else
        {
            update(2*node+1, m+1, right, poz, val);
        }

        aint[node] = aint[2*node] + aint[2*node+1];
    }
}


int query(int node, int left, int right, int qleft, int qright)
{
    if(qleft <= left && right <= qright)
    {
        return aint[node];
    }
    else
    {
        int m = (left + right) / 2;

        if(qright <= m)
        {
            return query(2*node, left, m, qleft, qright);
        }
        else if(m+1 <= qleft)
        {
            return query(2*node+1, m+1, right, qleft, qright);
        }

        return (query(2*node, left, m, qleft, qright) + query(2*node+1, m+1, right, qleft, qright));
    }
}

int main()
{
    in>>n>>m;

    for(int i = 1; i<=n; i++)
    {
        in>>v[i];
    }

    build(1, 1, n);

    int c, x, y;
    while(m--)
    {
        in>>c>>x>>y;

        if(c == 0)
        {
            update(1, 1, n, x, y);
        }
        else
        {
            out<<query(1, 1, n, x, y)<<'\n';
        }
    }

    return 0;
}