HDU 1166 敌兵布阵 线段树模板

HDU 1166 敌兵布阵 线段树模板

题目内容是线段树的单点修改和区间查询，可以用作线段树单点更新的模板。

[输入]
第一行一个整数T，表示有T组数据。
每组数据第一行一个正整数N（N<=50000）,表示敌人有N个工兵营地，接下来有N个正整数,第i个正整数ai代表第i个工兵营地里开始时有ai个人（1<=ai<=50）。
接下来每行有一条命令，命令有4种形式：
(1) Add i j,i和j为正整数,表示第i个营地增加j个人（j不超过30）
(2)Sub i j ,i和j为正整数,表示第i个营地减少j个人（j不超过30）;
(3)Query i j ,i和j为正整数,i<=j，表示询问第i到第j个营地的总人数;
(4)End 表示结束，这条命令在每组数据最后出现;
每组数据最多有40000条命令

[输出]
对第i组数据,首先输出“Case i:”和回车,
对于每个Query询问，输出一个整数并回车,表示询问的段中的总人数,这个数保持在int以内。

[样例输入]
1
10
1 2 3 4 5 6 7 8 9 10
Query 1 3
Add 3 6
Query 2 7
Sub 10 2
Add 6 3
Query 3 10
End

[样例输出]
Case 1:
6
33
59

AC代码：

#include <iostream>
#include <cstdio> 
#include <string>
#include <cstring>
using namespace std;
int ql,qr,x,d,temp[50005],a[50005*4];
void build(int o,int l,int r)
{
	if(l==r)
		a[o]=temp[l];
	else
	{
		int m=(l+r)/2;
		int ls=2*o,rs=2*o+1;
		build(ls,l,m);
		build(rs,m+1,r);
		a[o]=a[ls]+a[rs];
	}
}//建线段树 
int query(int o,int l,int r)
{
	if(ql<=l&&r<=qr)
		return a[o];
	else
	{
		int m=(l+r)/2;
		int ls=2*o,rs=2*o+1;
		int sum=0;
		if(ql<=m)
			sum+=query(ls,l,m);
		if(qr>=m+1)
			sum+=query(rs,m+1,r);
		return sum;
	}
}//查询 
void update(int o, int l, int r)
{
	if(l==r)
		a[o]+=d;
	else
	{
		int m=(l+r)/2;
		int ls=2*o,rs=2*o+1;
		if(x<=m)
			update(ls,l,m);
		else
			update(rs,m+1,r);
		a[o]=a[ls]+a[rs];
	}
}//修改 
int main()
{
	int t;
	cin>>t;
	for(int T=1;T<=t;T++)
	{
		int n;
		cin>>n;
		for(int i=1;i<=n;i++)
			cin>>temp[i];
		//build(1,1,n);
		memset(a,0,sizeof(a)); 
		for(int i=1;i<=n;i++)
		{
			x=i,d=temp[i];
			update(1,1,n);
		}
		printf("Case %d:\n",T);
		string s;
		int a,b;
		while(cin>>s)
		{
			if(s=="Query")
			{
				cin>>a>>b;
				ql=a,qr=b;
				printf("%d\n",query(1,1,n));
			}
			else if(s=="Add")
			{
				cin>>a>>b;
				x=a,d=b;
				update(1,1,n);
			}
			else if(s=="Sub")
			{
				cin>>a>>b;
				x=a,d=-b;//减去即加上负数 
				update(1,1,n);
			}
			else break;
		}
	}
	return 0;
}