Cod sursa(job #1399217)

Utilizator robertstrecheStreche Robert robertstreche Data 24 martie 2015 17:15:30
Problema Datorii Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.37 kb
#include <cstdio>

#define NMAX 15005
#define max(a,b) a>b?a:b

using namespace std;

long long val,tree[3*NMAX];
int n,tip,poz,st,dr,nr_query;

inline void update(int nod,int poz1,int poz2)
{
    if (poz1==poz2)tree[nod]=val;
    else
    {
        int m=(poz1+poz2)/2;
        if (poz<=m)update(2*nod,poz1,m);
        else update(2*nod+1,m+1,poz2);
        tree[nod]+=val;
    }
}
inline long long query(int nod,int poz1,int poz2)
{
    if (poz1>=st && poz2<=dr)return tree[nod];
    if (poz1>dr || poz2<st)return 0;
    int m=(poz1+poz2)/2;
    if (st<=m && dr>m)
    {
        int ma1=query(2*nod,poz1,m);
        int ma2=query(2*nod+1,m+1,poz2);
        return ma1+ma2;
    }
    if (st<=m)return query(2*nod,poz1,m);
    return query(2*nod+1,m+1,poz2);
}
int main()
{
    freopen("datorii.in","r",stdin);
    freopen("datorii.out","w",stdout);

    scanf("%d %d",&n,&nr_query);
    for (int i=1;i<=n;i++)
    {
        poz=i;
        scanf("%lld",&val);
        update(1,1,n);
    }
    for (int i=1;i<=nr_query;i++)
    {
        scanf("%d",&tip);
        if (tip==0)
        {
            scanf("%d %lld",&poz,&val);
            val=-val;
            update(1,1,n);
        }
        else
        {
            scanf("%d %d",&st,&dr);
            printf("%d\n",query(1,1,n));
        }
    }
    fclose(stdin);
    fclose(stdout);
}