Cod sursa(job #2379031)

Utilizator RaresLiscanLiscan Rares RaresLiscan Data 12 martie 2019 20:25:26
Problema Datorii Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.88 kb
#include <bits/stdc++.h>

using namespace std;

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

long long aib[15005];

void updateBIT (int idx, int val, int n)
{
    while (idx <= n) {
        aib[idx] += val;
        idx += (idx & (-idx));
    }
}

int read (int idx)
{
    int sum = 0;
    while (idx > 0) {
        sum += aib[idx];
        idx -= (idx & (-idx));
    }
    return sum;
}

void pay (int idx, int val, int n)
{
    while (idx <= n) {
        aib[idx] -= val;
        idx += (idx & (-idx));
    }
}

int main()
{
    int n, m;
    fin >> n >> m;
    int a, b, c, x;
    for (int i = 1; i <= n; i ++) {
        fin >> x;
        updateBIT(i, x, n);
    }
    for (int i = 1; i <= n; i ++) {
        fin >> a >> b >> c;
        if (a == 0) pay(b, c, n);
        else fout << read(c) - read(b - 1) << "\n";
    }
    return 0;
}