Cod sursa(job #2009237)

Utilizator GeoeyMexicanuBadita George GeoeyMexicanu Data 8 august 2017 23:39:31
Problema Datorii Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.25 kb
#include <iostream>
#include <fstream>
#define N 20010

using namespace std;

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

int arb[4*N+126],i,j,n,m,sum=0,st,dr,start,finish,val,x,cd,a,b,pos;
void update(int nod,int st,int dr)
{
    if(st==dr)
    {
        if(arb[nod]==0)
            arb[nod]=val;
        else
            arb[nod]=arb[nod]-val;
        return;
    }
    int mij=(st+dr)/2;
    if(pos<mij)
        update(2*nod,st,mij);
    else
        update(2*nod+1,mij+1,dr);
    arb[nod]=arb[2*nod]+arb[2*nod+1];
}
void query(int nod,int st,int dr)
{
    if(start<=st && dr<=finish)
    {
        sum+=arb[nod];
        return;
    }
    int mij=(st+dr)/2;
    if(start<=mij)
        query(2*nod,st,mij);
    if(mij<finish)
        query(2*nod+1,mij+1,dr);
}
int main()
{
    f>>n>>m;
    for(i=1;i<=n;i++)
    {
        f>>x;
        pos=i;
        val=x;
        update(1,1,n);
    }
    for(i=1;i<=m;i++)
    {
        f>>cd>>a>>b;
        if(cd==0)
        {
            pos=a;
            val=b;
            update(1,1,n);
        }
        else
        {
            start=a;
            finish=b;
            sum=0;
            query(1,1,n);
            g<<sum<<"\n";
        }
    }
}