Cod sursa(job #2086812)

Utilizator andreiomd1Onut Andrei andreiomd1 Data 12 decembrie 2017 15:40:40
Problema Datorii Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.07 kb
#include <fstream>
#define N 15000

using namespace std;
ifstream f("datorii.in");
ofstream g("datorii.out");
int arb[4*N+5];
int n,i,j,mo;
long int x;
int m;
int y,z;

void update (int poz, int val, int nod, int a, int b)
{
    if(a==b)
    {
        arb[nod]+=val;
        return;
    }
    else
    {
        int m=(a+b)/2;
        if(poz<=m)
            update(poz, val, 2*nod, a, m);
        else update(poz, val, 2*nod+1, m+1, b);

        arb[nod]=arb[2*nod]+arb[2*nod+1];
    }
}

int query (int nod, int qa, int qb, int st, int dr)
{
    int r1=0, r2=0;
    if(qa<=st&&qb>=dr)
        return arb[nod];
    m=(st+dr)/2;
    if(qa<=m)
        r1=query(2*nod, qa, qb, st, m);
    if(qb>m)
        r2=query(2*nod+1, qa, qb, m+1, dr);
    return (r1+r2);
}

int main()
{
    f>>n>>mo;
    for(i=1; i<=n; i++)
    {
        f>>x;
        update(i,x,1,1,n);
    }
    for(i=1; i<=mo; i++)
    {
        f>>x>>y>>z;
        if(x==1)
            g<<query(1, y, z, 1, n)<<'\n';
        else update(y, -z, 1, 1, n);
    }
    return 0;
}