敌兵布阵 HDU - 1166 线段树

敌兵布阵 HDU - 1166

input
第一行一个整数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条命令
output
对第i组数据,首先输出“Case i:”和回车,
对于每个Query询问，输出一个整数并回车,表示询问的段中的总人数,这个数保持在int以内。

这道题有区间查询，要么用树状数组，要么用线段树，介于这是在线段树专题的上的，我们就用线段树做，不过我觉得这个题少了线段树的灵魂lazy标记

算法实现

1. 单点修改（递归修改）
2. 线段查询

代码实现

#include<iostream>
#include<cstdio>
#include<string.h>
#include<string>
#define Maxn 50005

using namespace std;
struct Node
{
	int l,r;
	int sum;
	Node()
	{
		l = r = sum = 0;
	}
}sgt[Maxn * 4]; //不要开的太小了
void empty_sgt(Node *sgt)
{
	for(int i = 1;i <= Maxn * 4 - 1;i++)
	{
		sgt[i].l = sgt[i].r = sgt[i].sum = 0;
	}
}
int per[50005];
char d[10];
void push_up(int n)//向上维护总数
{
	sgt[n].sum = (sgt[n << 1].sum + sgt[n << 1 | 1].sum);
}
void build_tree(int l,int r,int n) //树的搭建
{

	sgt[n].l = l;
	sgt[n].r = r;
	if(l == r)
	{
		sgt[n].sum = per[l];
		return;
	}
	build_tree(l,(l + r) >> 1,n << 1);
	build_tree(((l + r) >> 1) + 1 , r,n << 1 | 1);
	push_up(n);
}
void Add(int i,int j,int n)//递归修改，记得最后要向上维护
{
	int l = sgt[n].l;
	int r = sgt[n].r;
	if(sgt[n].l == i && sgt[n].r == i)
	{
		sgt[n].sum += j;
		return;
	}
	else
	{
		int mid = (l + r) >> 1;
		if(i <= mid)Add(i,j,n << 1);
		else Add(i,j,n << 1 | 1);
		push_up(n); //不要忘 
	}
}

void Sub(int i,int j,int n)
{
	int l = sgt[n].l;
	int r = sgt[n].r;
	if(sgt[n].l == i && sgt[n].r == i)
	{
		sgt[n].sum -= j;
		return;
	}
	else
	{
		int mid = (l + r) >> 1;
		if(i <= mid)Sub(i,j,n << 1);
		else Sub(i,j,n << 1 | 1);
		push_up(n);//不要忘 
	}
}

int Query(int i,int j,int n)//递归查询，分成3种条件
{
	int tot = 0;
	int l = sgt[n].l;
	int r = sgt[n].r;
	if(l >= i && r <= j) //如果说一整个区间都在要查询的中，直接返回就可以了
	{
		tot += sgt[n].sum;
		return tot;
	}
	int mid = (l + r) >> 1;
	if(j > mid)tot += Query(i,j,n << 1 | 1); //如果区间的左子节点包含了查询区间 就递归查询左子节点
	if(i <= mid)tot += Query(i,j,n << 1);//如果区间的右子节点包含了查询区间 就递归查询右子节点
	
	return tot;
}
int main()
{
	int n,cnt;
	cin >> cnt;
	for(int k = 1;k <= cnt;k++)
	{
		cout << "Case" << ' ' << k <<':'<< endl;
		cin >> n;
		memset(per,0,sizeof(per));
		empty_sgt(sgt);
		for(int i = 1;i <= n;i++)
		{
			scanf("%d", &per[i]);
		}
		build_tree(1,n,1);
		cin >> d;
		while(1)
		{
			int flag = 0;
				switch(d[0])
			{
				case 'Q':
					{
						int i,j;
						cin >> i >> j;
						cout << Query(i,j,1) << endl;
						memset(d,0,sizeof(d));
						cin >> d;
						break;
					}
				case 'A':
					{
						int i,j;
						cin >> i >> j;
						Add(i,j,1);
						memset(d,0,sizeof(d));
						cin >> d;
						break;
					}
				case 'S':
					{
						int i,j;
						cin >> i >> j;
						Sub(i,j,1);
						memset(d,0,sizeof(d));
						cin >> d;
						break;
					}
				case 'E': 
					{
						flag = 1;
						break;
					}
			}
			if(flag == 1)break;
		}
	}
	return 0;
}