Cod sursa(job #1825007)

Utilizator SlevySlevoaca Stefan-Gabriel Slevy Data 8 decembrie 2016 17:45:07
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.92 kb
#include <iostream>
#include <fstream>
#define lsb(x) (x & (-x))
#define NMAX 15001

using namespace std;

ifstream in("datorii.in");
ofstream out("datorii.out");
int AIB[NMAX], n, m;

void Update_add(int poz, int key)
{
    for (int i = poz; i <= n; i += lsb(i))
        AIB[i] += key;
}

void Update_sub(int poz, int key)
{
    for (int i = poz; i <= n; i += lsb(i))
        AIB[i] -= key;
}

int Query(int poz)
{
    int sum = 0;
    for (int i = poz; i >= 1; i -= lsb(i))
        sum += AIB[i];
    return sum;
}

int main()
{
    in >> n >> m;
    int x, y, z;
    for (int i = 1; i <= n; i++)
    {
        in >> x;
        Update_add(i, x);
    }
    for (int i = 1; i <= m; i++)
    {
        in >> x >> y >> z;
        if (x == 0)
            Update_sub(y, z);
        else
            out << Query(z) - Query(y - 1) << "\n";
    }
    in.close();
    out.close();
    return 0;
}