Cod sursa(job #1656187)

Utilizator robertkarolRobert Szarvas robertkarol Data 18 martie 2016 21:26:47
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.06 kb
#include <iostream>
#include <fstream>
using namespace std;
ifstream fin("datorii.in");
ofstream fout("datorii.out");
int n,m,i,poz,nr,a,b,t,v[15001],arb[65000],s;
void update(int nod, int ls, int ld)
{
    if(ls==ld)
    {
        arb[nod]=nr;
        return;
    }
    int mij=(ls+ld)/2;
    if(poz<=mij) update(2*nod,ls,mij);
    else update(2*nod+1,mij+1,ld);
    arb[nod]=arb[2*nod]+arb[2*nod+1];
}
void suma(int nod, int ls, int ld)
{
    if(a<=ls&&ld<=b)
    {
        s+=arb[nod];
        return;
    }
    int mij=(ls+ld)/2;
    if(a<=mij) suma(2*nod,ls,mij);
    if(b>mij) suma(2*nod+1,mij+1,ld);
}
int main()
{
    fin>>n>>m;
    for(i=1;i<=n;i++)
    {
        fin>>v[i];
        poz=i; nr=v[i];
        update(1,1,n);
    }
    for(i=1;i<=m;i++)
    {
        fin>>t>>a>>b;
        if(t==0)
        {
            poz=a; nr=v[poz]-b;
            v[poz]-=b;
            update(1,1,n);
        }
        else
        {
            s=0;
            suma(1,1,n);
            fout<<s<<"\n";
        }
    }
    return 0;
}