Cod sursa(job #1506635)

Utilizator zertixMaradin Octavian zertix Data 20 octombrie 2015 20:50:57
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.58 kb
#include <iostream>
#include <cstdio>
using namespace std;

int n,m,x,pos,val,arb[60080],z,ach;
int fin,inc,s;


void update(int ,int ,int );
void update_zi(int ,int ,int );
void query(int ,int ,int );

int main()
{
    freopen("datorii.in","r",stdin);
    freopen("datorii.out","w",stdout);
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;++i)
    {
        scanf("%d",&x);
        pos=i;
        val=x;
        update(1,1,n);
    }

    for (int i=1;i<=m;++i)
    {
        scanf("%d%d%d",&x,&z,&ach);
        if (x==0)
        {
            pos=z;
            update_zi(1,1,n);
        }
        else
        {
            inc=z;
            fin=ach;
            s=0;
            query(1,1,n);
            printf("%d\n",s);
        }
    }

    return 0;
}
void update(int nod, int st ,int dr)
{
    if (st == dr)
    {
        arb[nod]=val;
        return;
    }
    int div=(st+dr)/2;
    if (pos<=div)
        update(2*nod,st,div);
    else
        update(2*nod+1,div+1,dr);

    arb[nod]=arb[2*nod]+arb[2*nod+1];
}

void update_zi(int nod, int st ,int dr)
{
    if (st == dr)
    {
        arb[nod]-=ach;
        return;
    }
    int div=(st+dr)/2;
    if (pos<=div)
        update_zi(2*nod,st,div);
    else
        update_zi(2*nod+1,div+1,dr);

    arb[nod]=arb[2*nod]+arb[2*nod+1];
}
void query(int nod , int st ,int dr)
{
    if (inc<=st && dr<=fin)
    {
        s+=arb[nod];
        return;
    }
    int div=(st+dr)/2;
    if ( inc<=div)
        query(2*nod,st,div);
    if (div<fin)
        query(2*nod+1,div+1,dr);
}