【题解】「HDU1166」敌兵布阵(线段树)

题面

【题目描述】
有$n$个营地，已知每个营地的人数，有四条命令：
$(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\le j$，表示询问第$i$到第$j$个营地的总人数;
$(4)End$ 表示结束，这条命令在每组数据最后出现;
【输入格式】
第一行一个整数$T$，表示有$T$组数据。
每组数据第一行一个正整数N（$N\le 50000$）,表示敌人有N个工兵营地，接下来有$N$个正整数,第$i$个正整数$a_i$代表第$i$个工兵营地里开始时有$a_i$个人（$1\le a_i\le 50$）。
接下来每行有一条命令，最多$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

算法分析

线段树模板题目。
单点修改，区间查询。

参考程序

#include<bits/stdc++.h>
#define N 50010      
using namespace std;
int T,n,a[N],s[4*N]; //4倍n
void built(int k,int l,int r)				//建线段树
{
    if(l==r)    {s[k]=a[l];return;}
    int mid=(l+r)/2;
    built(2*k,l,mid);
    built(2*k+1,mid+1,r);
    s[k]=s[k*2]+s[k*2+1];
}
void update(int k,int l,int r,int x,int v)		//单点更新
{
    if(l==r&&l==x){s[k]+=v;return;}				//找到单点修改
    if(x<l||x>r) return;
    int mid=(l+r)/2;
    if(l<=x&&x<=mid) update(k*2,l,mid,x,v);
    if(mid+1<=x&&x<=r) update(k*2+1,mid+1,r,x,v);
    s[k]=s[k*2]+s[k*2+1];					//回溯，维护区间和
}
int ask(int k,int l,int r,int x,int y)		//区间查询
{
    if(y<l||x>r) return 0;					//不相交
    if(x<=l&&r<=y) return s[k];				//包含
    int mid=(l+r)/2;
    return ask(k*2,l,mid,x,y)+ask(k*2+1,mid+1,r,x,y);
}
int main(){
    scanf("%d",&T);
    int count=1;
    while(count<=T)
    {
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
            scanf("%d",&a[i]);
        built(1,1,n);
        char s[10];
        int x,y;
        printf("Case %d:\n",count);
        while(1)
        {
            scanf("%s",s);
            if(s[0]=='E') break;
            scanf("%d%d",&x,&y);
            if(s[0]=='A') update(1,1,n,x,y);
            if(s[0]=='S') update(1,1,n,x,-y);
            if(s[0]=='Q') printf("%d\n",ask(1,1,n,x,y));
        }
        count++;
     } 
    return 0;
}