Cod sursa(job #434067)

Utilizator cosmin79Carabet Cosmin Andrei cosmin79 Data 4 aprilie 2010 23:30:48
Problema Datorii Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.94 kb
#include <stdio.h>
#define NMAX 15005
#define LMAX 1<<15
int n,m,init[NMAX],arb[LMAX],x,y;
void cstr(int st,int dr,int nod)
{
	if (st==dr)
	{
		init[st]=nod;
		return ;
	}
	int mij=(st+dr)>>1;
	cstr(st,mij,nod<<1);
	cstr(mij+1,dr,(nod<<1)+1);
}
void update(int ind,int val)
{
	int nod=init[ind];
	arb[nod]+=val;
	nod>>=1;
	for ( ; nod; )
	{
		arb[nod]=arb[nod<<1]+arb[(nod<<1)+1];
		nod>>=1;
	}
}
int query(int st,int dr,int nod)
{
	if (x<=st && dr<=y)
		return arb[nod];
	if (x>dr || y<st)
		return 0;
	int mij=(st+dr)/2;
	return query(st,mij,nod<<1)+query(mij+1,dr,(nod<<1)+1);
}
int main()
{
	freopen("datorii.in","r",stdin);
	freopen("datorii.out","w",stdout);
	scanf("%d%d",&n,&m);
	cstr(1,n,1);
	int i,tip;
	for (i=1; i<=n; i++)
	{
		scanf("%d",&x);
		update(i,x);
	}
	while (m--)
	{
		scanf("%d%d%d",&tip,&x,&y);
		if (!tip)
			update(x,-y);
		else
			printf("%d\n",query(1,n,1));
	}
	return 0;
}