Cod sursa(job #1490275)

Utilizator raducostacheRadu Costache raducostache Data 23 septembrie 2015 08:48:22
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.01 kb
#include <fstream>

#define NMAX 20000

using namespace std;

ifstream f("datorii.in");
ofstream g("datorii.out");

int sol[NMAX] , x , y , n , m , tip;

void update1(int poz , int val);
void update2(int poz , int val);
int query(int poz);

int main()
{
    f >> n >> m;
    for(int i = 1 ; i <= n ; ++i){
        f >> x;
        update1(i , x);
    }

    for(int i = 1 ; i <= m ; ++i){
        f >> tip >> x >> y;
        if(!tip){
            update2(x , y);
        }
        else{
            g << query(y) - query(x - 1) << '\n';
        }
    }
    return 0;
}
void update1(int poz , int val){
    while(poz <= n){
        sol[poz] += val;
        poz += (poz & (poz ^ (poz - 1)));
    }
}


void update2(int poz , int val){
    while(poz <= n){
        sol[poz] -= val;
        poz += (poz & (poz ^ (poz - 1)));
    }
}

int query(int poz){
    int sum = 0;
    while(poz > 0){
        sum += sol[poz];
        poz -= (poz & (poz ^ (poz - 1)));
    }
    return sum;
}