【題解】「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\leq j$，表示詢問第$i$到第$j$個營地的總人數;
$(4)End$ 表示結束，這條命令在每組數據最後出現;
【輸入格式】
第一行一個整數$T$，表示有$T$組數據。
每組數據第一行一個正整數N（$N \leq 50000$）,表示敵人有N個工兵營地，接下來有$N$個正整數,第$i$個正整數$a_i$代表第$i$個工兵營地裏開始時有$a_i$個人（$1 \leq a_i \leq 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;
}