Cod sursa(job #2223482)

Utilizator caesar2001Stoica Alexandru caesar2001 Data 20 iulie 2018 13:50:40
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.36 kb
#include <bits/stdc++.h>

using namespace std;
ifstream in("datorii.in");
ofstream out("datorii.out");

const int NMAX = 15005;
int aint[4 * NMAX];

void update(int x, int pos, int node, int left, int right) {
    if(left == right)
        aint[node] += x;
    else {
        int mid = (left + right) / 2;
        if(pos <= mid)
            update(x, pos, node * 2, left, mid);
        else
            update(x, pos, node * 2 + 1, mid + 1, right);
        aint[node] = aint[node * 2] + aint[node * 2 + 1];
    }
}

int query(int from, int to, int node, int left, int right) {
    if(from <= left && right <= to)
        return aint[node];
    else {
        int mid = (left + right) / 2, s = 0;
        if(from <= mid)
            s += query(from, to, node * 2, left, mid);
        if(mid < to)
            s += query(from, to, node * 2 + 1, mid + 1, right);
        return s;
    }
}

int main()
{
    ios::sync_with_stdio(false);
    in.tie(0);
    out.tie(0);
    int n, m;
    in >> n >> m;
    for(int i = 1; i <= n; i ++) {
        int x;
        in >> x;
        update(x, i, 1, 1, n);
    }
    for(int i = 1; i <= m; i ++) {
        int test, t, v;
        in >> test >> t >> v;
        if(test == 0)
            update((-1) * v, t, 1, 1, n);
        else
            out << query(t, v, 1, 1, n) << "\n";
    }
    return 0;
}