Cod sursa(job #1579557)

Utilizator claudiuarseneClaudiu Arsene claudiuarsene Data 24 ianuarie 2016 21:03:05
Problema Datorii Scor 100
Compilator cpp Status done
Runda runda_de_verificare1 Marime 1.51 kb
#include <fstream>
#define tata(x) (x)/2
#define fst(x) (x)<<1
#define fdr(x) ((x)<<1)+1
using namespace std;


ifstream f("datorii.in");
ofstream g("datorii.out");
int op,val1,val2, arbint[400001],a[100001],n,m;

void update(int poz,int st, int dr, int vpoz, int val) {
    if(st == dr) {
        arbint[poz] -= val;
        return;
    }
    int mijl = (st + dr) / 2;
    if(vpoz > mijl) {
        update(fdr(poz),mijl+1,dr,vpoz,val);
    }
    else {
        update(fst(poz),st,mijl,vpoz,val);
    }
    arbint[poz] = arbint[fst(poz)]+arbint[fdr(poz)];
}

int query(int poz, int st, int dr, int a , int b) {
    if(st >= a && dr <= b) {
        return arbint[poz];
    }
    int mijl = (st + dr)>>1;
    int suma = 0;
    if(b > mijl)
        suma = query(fdr(poz),mijl+1,dr,a,b);
    if(a <= mijl)
        suma += query(fst(poz),st,mijl,a,b);
    return suma;
}

void create(int poz, int st, int dr) {
    if(st == dr) {
        arbint[poz] = a[st];
        return;
    }
    int mijl = (st + dr)>>1;
    create(fst(poz),st,mijl);
    create(fdr(poz),mijl+1,dr);
    arbint[poz] = arbint[fst(poz)]+arbint[fdr(poz)];
}

void citire() {
    f >> n >> m;
    for(int i = 1; i <= n; i++) {
        f >> a[i];
    }

    create(1,1,n);
    for(int i = 0; i < m; i++) {

         f >> op >> val1 >> val2;
         if(op == 1)
            g << query(1,1,n,val1,val2) << "\n";
        else
            update(1,1,n,val1,val2);
    }
    f.close();
}

int main() {
    citire();
}