Cod sursa(job #1267162)

Utilizator andi12Draghici Andrei andi12 Data 19 noiembrie 2014 16:57:26
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.34 kb
#include <cstdio>

using namespace std;
const int N=100001;
int val,poz,start,fin,sum;
int arb[4*N+66];
void update(int nod,int st,int dr)
{
    if(st==dr)
    {
        arb[nod]=arb[nod]-val;
        return ;
    }
    else
    {
        int mid=(st+dr)/2;
        if(poz<=mid)
            update(nod*2,st,mid);
        else
            update(nod*2+1,mid+1,dr);
    }
    arb[nod]=arb[2*nod]+arb[2*nod+1];
}
void ras(int nod,int st,int dr)
{
    if(start<=st && dr<=fin)
    {
        sum=sum+arb[nod];
        return ;
    }
    else
    {
        int mid=(st+dr)/2;
        if(start<=mid)
            ras(nod*2,st,mid);
        if(fin>mid)
            ras(nod*2+1,mid+1,dr);
    }
}
int main()
{
    FILE *in,*out;
    in=fopen("datorii.in","r");
    out=fopen("datorii.out","w");
    int n,m,i,j,x,c,a,b;
    fscanf(in,"%d%d",&n,&m);
    for(i=1;i<=n;i++)
    {
        fscanf(in,"%d",&x);
        val=-x;
        poz=i;
        update(1,1,n);
    }
    for(i=1;i<=m;i++)
    {
        fscanf(in,"%d%d%d",&c,&a,&b);
        if(c==0)
        {
            poz=a;
            val=b;
            update(1,1,n);
        }
        else
        {
            sum=0;
            start=a;
            fin=b;
            ras(1,1,n);
            fprintf(out,"%d\n",sum);
        }
    }
    return 0;
}