敌兵布阵（线段树）

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;
}