敌兵布阵 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标记
算法实现
- 单点修改(递归修改)
- 线段查询
代码实现
#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;
}