Cod sursa(job #1867144)

Utilizator geralt_of_riviajohn nathalis geralt_of_rivia Data 3 februarie 2017 19:40:02
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.29 kb
#include <fstream>

using namespace std;
ifstream cin ("datorii.in");
ofstream cout ("datorii.out");
int n,m,poz,c,arb[50010],a,b;
void add (int st,int dr,int nod)
{
    if(st==poz && dr==poz)
        arb[nod]=c;
    else
    {
        int mij=(st+dr)/2;
        if(poz<=mij) add(st,mij,nod*2);
        else add(mij+1,dr,nod*2+1);
        arb[nod]=arb[nod*2]+arb[nod*2+1];
    }
}
void up_date (int st,int dr,int nod)
{
    if(st==poz && dr==poz)
        arb[nod]-=c;
    else
    {
        int mij=(st+dr)/2;
        if(poz<=mij) up_date(st,mij,nod*2);
        else up_date(mij+1,dr,nod*2+1);
        arb[nod]=arb[nod*2]+arb[nod*2+1];
    }
}
void read ()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        cin>>c; poz=i;
        add(1,n,1);
    }
}
int query (int st,int dr,int nod)
{
    if(st>=a && dr<=b)
        return arb[nod];
    int mij=(st+dr)/2;
    int h1=0,h2=0;
    if(a<=mij) h1=query(st,mij,nod*2);
    if(b>mij) h2=query(mij+1,dr,nod*2+1);
    return h1+h2;
}
void solve_here ()
{ int p;
    for(int i=1;i<=m;i++)
    {
        cin>>p>>a>>b;
        if(p==0) { poz=a; c=b; up_date(1,n,1); }
        else cout<<query(1,n,1)<<"\n";
    }
}
int main()
{
    read();
    solve_here();
    cin.close();
    cout.close();
    return 0;
}