Cod sursa(job #1045847)

Utilizator alexpascadiAlexandru Pascadi alexpascadi Data 2 decembrie 2013 09:54:43
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.32 kb
#include <stdio.h>

using namespace std;

const int N=15001;
int v[N],t[4*N];
int n,m,suma,a,b,val,poz;

void query(int p, int st, int dr)
{
    if(a<=st && dr<=b)
    {
        suma+=t[p];
        return;
    }
    int m=(st+dr)>>1;
    if(a<=m)
    {
        query(p*2,st,m);
    }
    if(m+1<=b)
    {
        query(p*2+1,m+1,dr);
    }
}

void update(int p, int st, int dr)
{
    if(st==dr)
    {
        t[p]=val;
    }
    else
    {
        int m=st+dr; m>>=1;
        if(poz<=m)
        {
            update(2*p,st,m);
        }
        else
        {
            update(2*p+1,m+1,dr);
        }
        t[p]=t[2*p]+t[2*p+1];
    }
}

int main()
{
    FILE *in,*out;
    in=fopen("datorii.in","r");
    out=fopen("datorii.out","w");
    int i,x;
    fscanf(in,"%d%d",&n,&m);
    for(i=1;i<=n;i++)
    {
        fscanf(in,"%d",&v[i]); val=v[i]; poz=i;
        update(1,1,n);
    }
    for(i=1;i<=m;i++)
    {
        fscanf(in,"%d",&x);
        if(x==0)
        {
            fscanf(in,"%d%d",&poz,&val);
            v[poz]-=val;
            val=v[poz];
            update(1,1,n);
        }
        else
        {
            fscanf(in,"%d%d",&a,&b);
            suma=0;
            query(1,1,n);
            fprintf(out,"%d\n",suma);
        }
    }
    return 0;
}