Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 65536/32768 K (Java/Others) Total Submission(s): 62044 Accepted Submission(s): 26225
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以内。
样例输入:
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
代码如下:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
struct Node
{
int rnode;
int lnode;
int SUM;
}tree[50000*4];
int t,n,value[50005],a,b,m;
/*
int t:为样例个数
int n:为区间大小
int value:为区间节点初始值
int a:为左边界|要修改节点的下标
int b:为右边界|要修改节点的权值
int m:为Case 数
*/
string order;//命令
int Sum;//所求区间权值总和
/*
函数功能:建立线段树
函数名称:InitTree
参 数:int l 为左边界
int r 为右边界
int nodeNum 为根节点下标
返 回 值 :无
*/
void InitTree(int l,int r,int nodeNum)
{
tree[nodeNum].lnode=l;//最左孩子下标
tree[nodeNum].rnode=r;//最右孩子下标
if(l==r) //如果相等则为原始数据节点
tree[nodeNum].SUM=value[l];
else
{
InitTree(l,(l+r)/2,2*nodeNum);//左孩子
InitTree((l+r)/2+1,r,2*nodeNum+1);//左孩子
tree[nodeNum].SUM=tree[nodeNum*2].SUM+tree[nodeNum*2+1].SUM;//父亲权值为左右孩子权值之和
}
}
/*
函数功能:更新区间权值
函数名称:Query
参 数:int l 为左边界
int r 为右边界
int nodeNum 为根节点下标
返 回 值 :无
*/
void Query(int l,int r,int nodeNum)
{
if(l<=tree[nodeNum].lnode&&r>=tree[nodeNum].rnode)//子区间
Sum+=tree[nodeNum].SUM;
else
{
if(r<=(tree[nodeNum].lnode+tree[nodeNum].rnode)/2) //恰好范围在左子树
Query(l,r,nodeNum*2);
else if(l>=(tree[nodeNum].lnode+tree[nodeNum].rnode)/2+1) //恰好范围在右子树
Query(l,r,2*nodeNum+1);
else //所有范围左右子树都有
{
Query(l,r,nodeNum*2);
Query(l,r,nodeNum*2+1);
}
}
}
/*
函数功能:线段树 增加节点的权值
函数名称:Add
参 数:int que 为要增加权值的节点
int val 为增加的权值大小
int nodeNum 为根节点下标
返 回 值 :无
*/
void Add(int que,int val,int nodeNum)
{
tree[nodeNum].SUM+=val; //为所递归到的根节点都添加该权值
if(tree[nodeNum].rnode==tree[nodeNum].lnode) // 直到找到该节点为止
return ;
if(que<=(tree[nodeNum].rnode+tree[nodeNum].lnode)/2) //在左孩子
Add(que,val,2*nodeNum);
else //在右孩子
Add(que,val,2*nodeNum+1);
}
/*
函数功能:线段树 减少节点的权值
函数名称:Sub
参 数:int que 为要减少权值的节点
int val 为减少的权值大小
int nodeNum 为根节点下标
返 回 值 :无
*/
void Sub(int que,int val,int nodeNum)
{
tree[nodeNum].SUM-=val; //为所递归到的根节点都减少该权值
if(tree[nodeNum].rnode==tree[nodeNum].lnode) // 直到找到该节点为止
return ;
if(que<=(tree[nodeNum].rnode+tree[nodeNum].lnode)/2) //在左孩子
Sub(que,val,2*nodeNum);
else //在右孩子
Sub(que,val,2*nodeNum+1);
}
int main()
{
scanf("%d",&t);
while(t--)
{
//memset(value,0,sizeof(value));
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%d",&value[i]);
InitTree(1,n,1);
printf("Case %d:\n",++m);
while(1)
{
scanf("%s",&order[0]);
if(order[0]=='E')//结束该组
break;
if(order[0]=='Q')//执行输出综合
{
scanf("%d %d",&a,&b);
Sum=0;
Query(a,b,1);
printf("%d\n",Sum);
}
if(order[0]=='A')//执行添加
{
scanf("%d %d",&a,&b);
Add(a,b,1);
}
if(order[0]=='S')//执行减少
{
scanf("%d %d",&a,&b);
Sub(a,b,1);
}
}
}
return 0;
}