Cod sursa(job #52473)

Utilizator pustiuRadu Zaharia pustiu Data 18 aprilie 2007 22:36:45
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.64 kb
#include<stdio.h>

FILE *in = fopen ("datorii.in", "rt");
FILE *out = fopen ("datorii.out", "wt");

int N,M,q[15000],K=0;

int A[35000];
/*
void construieste (int k, int x, int y)
{
    A[k].s=x;
    A[k].d=y;
    A[k].v=0;
    if(x!=y)
    {
    	construieste (2*k,x,(x+y)/2);
    	construieste (2*k+1,(x+y)/2+1,y);
    }
}
*/

void adauga (int k,int s, int d,int x, int y)
{
    A[k]+=y;
    if(s!=d)
    {
    	if(x<=(s+d)/2)
    	{
    		adauga(2*k,s,(s+d)/2,x,y);
    	}
    	else
    	{
    		adauga(2*k+1,(s+d)/2+1,d,x,y);
    	}
    }
}

void scade (int k, int s, int d, int x, int y)
{
    A[k]-=y;
    if(s!=d)
    {
    	if(x<=(s+d)/2)
    	{
    		scade(2*k,s,(s+d)/2,x,y);
  		}
    	else
    	{
    		scade(2*k+1,(s+d)/2+1,d,x,y);
    	}
    }
}

void query(int k,int s, int d,int x, int y)
{
	if(x<=s && y>=d)
    {
    	K+=A[k];
    }
    else
    {
    	int z=(s+d)/2;
        if(y>z)
        	query(k*2+1,(s+d)/2+1,d,x,y);
        if(x<=z)
        	query(k*2,s,(s+d)/2,x,y);
    }
}

void citire ()
{
    fscanf (in, "%d %d", &N, &M);
	//construieste (1,1,N);
    for(int i=1;i<=N;i++)
    {
		int x;
        fscanf(in, "%d", &x);
        adauga(1,1,N,i,x);
    }
}

void rezolvare ()
{
    for(int i=1;i<=M;i++)
    {
    	int x,y,z;
        fscanf(in, "%d %d %d", &x, &y, &z);
        if(!x)
        {
        	scade(1,1,N,y,z);
        }
        else
        {
        	K=0;
            query(1,1,N,y,z);
            fprintf(out,"%d\n",K);
        }
    }
}

int main ()
{
	citire ();
    rezolvare ();
    fclose(in);
    fclose(out);
    return 0;
}