hdu 1166（线段树）

点击打开链接

都说是模版级别的。。

但是还是ＲＥ了两次啊（是因为输入的原因）。。。。

 

#include"stdio.h"

#include"string.h"

#define N 50011

struct tree

{

	int x,y,mid;

	int count;

}A[4*N];



void creat(int x,int y,int k)

{

	A[k].x=x;

	A[k].y=y;

	A[k].mid=(x+y)/2;

	A[k].count=0;

	if(x==y)return ;

	creat(x,A[k].mid,2*k);

	creat(A[k].mid+1,y,2*k+1);

	return;

}



void insert(int x,int y,int k)

{

	if(A[k].x==A[k].y&&A[k].x==x)

	{

		A[k].count+=y;

		return ;

	}

	if(x<=A[k].mid)insert(x,y,2*k);

	else insert(x,y,2*k+1);

	A[k].count=A[2*k].count+A[2*k+1].count;

	return ;

}



int search(int x,int y,int k)

{

	if(A[k].x==x&&A[k].y==y)

		return A[k].count;

	int ans=0;

	if(y<=A[k].mid)

		ans+=search(x,y,2*k);

	else if(x>A[k].mid)

		ans+=search(x,y,2*k+1);

	else

	{

		ans+=search(x,A[k].mid,2*k);

		ans+=search(A[k].mid+1,y,2*k+1);

	}

	return ans;

}

int main()

{

	int T,t;

	int n;

	int x,y,a;

	int i;

	char s[15];

	scanf("%d",&T);

	t=1;

	while(T--)

	{

		printf("Case %d:\n",t++);

		scanf("%d",&n);

		creat(1,n,1);

		for(i=1;i<=n;i++)

		{

			scanf("%d",&a);

			insert(i,a,1);

		}

		getchar();

		while(scanf("%s",s),s[0]!='E')

		{

			scanf("%d%d",&x,&y);

			if(s[0]=='A')insert(x,y,1);

			else if(s[0]=='S')insert(x,-y,1);

			else if(s[0]=='Q')printf("%d\n",search(x,y,1));

		}

	}

	return 0;

}