Cod sursa(job #1983928)

Utilizator dragos231456Neghina Dragos dragos231456 Data 22 mai 2017 21:09:22
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.32 kb
#include <iostream>
#include <fstream>
using namespace std;
ifstream f("datorii.in");
ofstream g("datorii.out");
int arb[60105],n,m,x,y,op,val;
long long sum;

void init(int nod,int lt,int rt,int pos)
{
    if(lt==rt)
    {
        arb[nod]=val;
        return;
    }
    int md=(lt+rt)/2;
    if(pos<=md) init(nod*2,lt,md,pos);
    else init(nod*2+1,md+1,rt,pos);

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

void update(int nod,int lt,int rt,int pos)
{
    if(lt==rt)
    {
        arb[nod]-=val;
        return;
    }
    int md=(lt+rt)/2;
    if(pos<=md) update(nod*2,lt,md,pos);
    else update(nod*2+1,md+1,rt,pos);

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

void query(int nod,int lt,int rt,int a,int b)
{
    if(a<=lt && rt<=b)
    {
        sum+=arb[nod];
        return;
    }
    if(lt==rt) return;
    int md=(lt+rt)/2;
    if(a<=md) query(nod*2,lt,md,a,b);
    if(b>md) query(nod*2+1,md+1,rt,a,b);
}

int main()
{
    f>>n>>m;
    for(int i=1;i<=n;++i)
    {
        f>>val;
        init(1,1,n,i);
    }
    for(int i=1;i<=m;++i)
    {
        f>>op>>x>>y;
        if(op==0)
        {
            val=y;
            update(1,1,n,x);
        }
        else
        {
            sum=0;
            query(1,1,n,x,y);
            g<<sum<<'\n';
        }
    }
    return 0;
}