Cod sursa(job #2415035)

Utilizator bogdanvladmihaiBogdan Vlad-Mihai bogdanvladmihai Data 25 aprilie 2019 14:25:59
Problema Datorii Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.06 kb
#include <bits/stdc++.h>

#define Nmax 15000
#define lsb(x) (x&(-x))

using namespace std;

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

int N, Q;

int AIB[Nmax + 5];

inline int query (int poz);
inline void update (int poz, int val);

int main()
{
    fin >> N >> Q;
    for (int i = 1; i <= N; ++i)
    {
        int x;
        fin >> x;
        update (i, x);
    }
    while (Q--)
    {
        int TASK;
        fin >> TASK;
        if (TASK)
        {
            int left, right;
            fin >> left >> right;
            fout << query (right) - query (left - 1) << "\n";
        }
        else
        {
            int val, poz;
            fin >> poz >> val;
            update (poz, -val);
        }
    }
    return 0;
}

inline void update (int poz, int val)
{
    while (poz <= N)
    {
        AIB[poz] += val;
        poz += lsb (poz);
    }
}

inline int query (int poz)
{
    int sum = 0;
    while (poz)
    {
        sum += AIB[poz];
        poz -= lsb (poz);
    }
    return sum;
}