Cod sursa(job #2068044)

Utilizator GeorgeCalinPetruta George-Calin GeorgeCalin Data 17 noiembrie 2017 09:33:36
Problema Datorii Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.4 kb
#include <fstream>
#define nmax 15002 * 2 + 1
using namespace std;
ifstream fin("datorii.in");
ofstream fout("datorii.out");

int v[nmax];
int val,poz,n,m,c,x,y,suma,inc,sf;

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

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

void query(int nod,int st,int dr)
{
    if(st>=inc&&dr<=sf)
    {
        suma+=v[nod];
        return;
    }
    if(st>sf||inc>dr)return;
    int mij=(st+dr)/2;
    query(nod*2,st,mij);
    query(nod*2+1,mij+1,dr);
}

int main()
{
    fin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        fin>>x;
        val=x;
        poz=i;
        update1(1,1,n);
    }
    for(int i=1;i<=m;i++)
    {
        fin>>c>>x>>y;
        if(c==0)
        {
            val=y;
            poz=x;
            update(1,1,n);
        }
        else
        {
            suma=0;
            inc=x;
            sf=y;
            query(1,1,n);
            fout<<suma<<"\n";
        }
    }
    return 0;
}