Cod sursa(job #1779877)

Utilizator Mircea_DonciuDonciu Mircea Mircea_Donciu Data 15 octombrie 2016 17:41:36
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.35 kb
#include <iostream>
#include <fstream>
using namespace std;
ifstream in("datorii.in");
ofstream out("datorii.out");
const int maxn = 15005;
int aint[maxn * 4];

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

int query(int nod, int st, int dr, int x, int y)
{
    if(dr < x || y < st)
        return 0;
    if(x <= st && dr <= y)
        return aint[nod];
    int mij = (st + dr) / 2;
    int ans1 = query(nod * 2, st, mij, x, y);
    int ans2 = query(nod * 2 + 1, mij + 1, dr, x, y);
    return ans1 + ans2;
}

int main()
{
    int n, m;
    in >> n >> m;
    for(int i = 1; i <= n; i++)
    {
        int x;
        in >> x;
        update(1, 1, n, i, -x);
    }
    for(int i = 1; i <= m; i++)
    {
        bool op;
        in >> op;
        if(op == 0)
        {
            int t, v;
            in >> t >> v;
            update(1, 1, n, t, v);
        }
        else
        {
            int p, q;
            in >> p >> q;
            out << query(1, 1, n, p, q) << "\n";
        }
    }
    return 0;
}