Cod sursa(job #1468559)

Utilizator MaligMamaliga cu smantana Malig Data 6 august 2015 13:05:47
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.57 kb
#include<iostream>
#include<fstream>
#include<cmath>
#include<algorithm>
#include<vector>
#include<bitset>
#include<cstring>
#include<queue>

#define ull unsigned long long
#define ll long long
#define pb push_back
#define FOR(a,b,c) for (int a=b;a<=c; ++a)
#define ROF(a,b,c) for (int a=b;a>=c; --a)

using namespace std;
ifstream f("datorii.in");
ofstream g("datorii.out");
int N,M;
int arb[46385];

void update(int,int,int,int,int);
int query(int,int,int,int,int);

int main()
{
    f>>N>>M;
    FOR (i,1,N) {
        int x;
        f>>x;
        update(1,1,N,i,x);
    }
    FOR (i,1,M) {
        int tip,a,b;
        f>>tip>>a>>b;
        if (!tip) {
            int da=-b;
            update(1,1,N,a,da);
        }
        else {
            g<<query(1,1,N,a,b)<<'\n';
        }
    }
    f.close();g.close();
    return 0;
}

void update(int nod,int st,int dr,int poz,int val) {
    if (st==dr) {
        arb[nod]+=val;
    }
    else {
        int mij=(st+dr)/2;
        if (poz<=mij) {
            update((nod<<1),st,mij,poz,val);
        }
        else {
            update((nod<<1)+1,mij+1,dr,poz,val);
        }
        arb[nod]=arb[(nod<<1)]+arb[(nod<<1)+1];
    }
}

int query(int nod,int st,int dr,int a,int b) {
    if (a<=st && dr<=b) {
        return arb[nod];
    }
    else {
        int mij=(st+dr)/2;
        int val=0;
        if (a<=mij) {
            val+=query((nod<<1),st,mij,a,b);
        }
        if (mij<b) {
            val+=query((nod<<1)+1,mij+1,dr,a,b);
        }
        return val;
    }
}