Cod sursa(job #2753158)

Utilizator pielevladutPiele Vladut Stefan pielevladut Data 21 mai 2021 11:58:34
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.97 kb
#include <bits/stdc++.h>

#define int long long

using namespace std;

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

int n, m;
int v[15151];
int aib[15151];

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

int query(int poz)
{
    int ans = 0;
    while(poz != 0)
    {
        ans += aib[poz];
        poz = poz - (poz & (-poz));
    }
    return ans;
}

int32_t main()
{
    fin >> n >> m;
    for(int i = 1; i <= n; i ++)
    {
        fin >> v[i];
        update(i, v[i]);
    }
    for(int i = 1; i <= m; i ++)
    {
        int op;
        fin >> op;
        if(!op)
        {
            int poz,val;
            fin >> poz >> val;
            update(poz, -val);
        }
        else
        {
            int st, dr;
            fin >> st >> dr;
            fout << query(dr) - query(st - 1) << '\n';
        }
    }
    return 0;
}