Cod sursa(job #1850976)

Utilizator geo_furduifurdui geo geo_furdui Data 19 ianuarie 2017 08:41:59
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.32 kb
#include <fstream>

using namespace std;
ifstream cin ("datorii.in");
ofstream cout ("datorii.out");
int n,m,key,poz,arb[350001],a,b;
void up_date (int st,int dr,int nod)
{
    if(st>=poz && dr<=poz)
        arb[nod]=key;
    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 up_date_minus (int st,int dr,int nod)
{
    if(st>=poz && dr<=poz)
        arb[nod]-=key;
    else
    {
        int mij=(st+dr)/2;
        if(poz<=mij) up_date_minus(st,mij,nod*2);
        else up_date_minus(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>>key,poz=i,up_date(1,n,1);
}
int ask_me (int st,int dr,int nod)
{
    if(a<=st && b>=dr)
        return arb[nod];
    int mij=(st+dr)/2;
    int x1=0,x2=0;
    if(a<=mij) x1=ask_me(st,mij,nod*2);
    if(b>mij) x2=ask_me(mij+1,dr,nod*2+1);
    return x1+x2;
}
void solve_here ()
{ int p;
    for(int i=1;i<=m;i++)
    {
        cin>>p>>a>>b; poz=a; key=b;
        if(p==0) up_date_minus(1,n,1);
        else cout<<ask_me(1,n,1)<<"\n";
    }
}
int main()
{
    read();
    solve_here();
    cin.close();
    cout.close();
    return 0;
}