HDU - 1166 敵兵佈陣 線段樹

敵兵佈陣

C國的死對頭A國這段時間正在進行軍事演習，因此C國間諜頭子Derek和他手下Tidy又開始忙乎了。A國在海岸線沿直線佈置了N個工兵營地,Derek和Tidy的任務就是要監視這些工兵營地的活動狀況。因爲採起了某種先進的監測手段，因此每一個工兵營地的人數C國都掌握的一清二楚,每一個工兵營地的人數都有可能發生變更，可能增長或減小若干人手,但這些都逃不過C國的監視。 
中央情報局要研究敵人究竟演習什麼戰術,因此Tidy要隨時向Derek彙報某一段連續的工兵營地一共有多少人,例如Derek問:「Tidy,立刻彙報第3個營地到第10個營地共有多少人!」Tidy就要立刻開始計算這一段的總人數並彙報。但敵兵營地的人數常常變更，而Derek每次詢問的段都不同，因此Tidy不得不每次都一個一個營地的去數，很快就精疲力盡了，Derek對Tidy的計算速度愈來愈不滿:"你個死肥仔，算得這麼慢，我炒你魷魚!」Tidy想：「你本身來算算看，這可真是一項累人的工做!我巴不得你炒我魷魚呢!」無奈之下，Tidy只好打電話向計算機專家Windbreaker求救,Windbreaker說：「死肥仔，叫你平時作多點acm題和看多點算法書，如今嚐到苦果了吧!」Tidy說："我知錯了。。。"但Windbreaker已經掛掉電話了。Tidy很苦惱，這麼算他真的會崩潰的，聰明的讀者，你能寫個程序幫他完成這項工做嗎？不過若是你的程序效率不夠高的話，Tidy仍是會受到Derek的責罵的. 

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之內。 
Sample Input

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 

Sample Output

Case 1:
6
33
59


在普通枚舉超時的狀況下，選擇線段樹處理區間問題。通常分爲點/區間更新，點/區間查詢幾類。一維數組存樹tree[4*n+1]，採用二分枚舉。

#include<stdio.h>

struct{
    int l,r,sum;
}tree[200005];

int r[50005],SUM;

void make(int left,int right,int num)
{
    tree[num].l=left;
    tree[num].r=right;
    if(left==right)
        tree[num].sum=r[left];
    else{
        make(left,(left+right)/2,2*num);
        make((left+right)/2+1,right,2*num+1);
        tree[num].sum=tree[2*num].sum+tree[2*num+1].sum;
    }
}

void query(int left,int right,int num)
{
    if(left<=tree[num].l&&right>=tree[num].r) SUM+=tree[num].sum;
    else{
        if(right<=(tree[num].l+tree[num].r)/2)
            query(left,right,2*num);
        else if(left>=(tree[num].l+tree[num].r)/2+1)
            query(left,right,2*num+1);
        else{
            query(left,right,2*num);
            query(left,right,2*num+1);
        }
    } 
}

void add(int x,int y,int num)
{
    tree[num].sum+=y;
    if(tree[num].l==tree[num].r) return;
    if(x>(tree[num].l+tree[num].r)/2)
        add(x,y,2*num+1);
    else
        add(x,y,2*num);
}

void sup(int x,int y,int num)
{
    tree[num].sum-=y;
    if(tree[num].l==tree[num].r) return;
    if(x>(tree[num].l+tree[num].r)/2)
        sup(x,y,2*num+1);
    else
        sup(x,y,2*num);
}

int main()
{
    int t,i,a,b,n,m=0;
    char s[10];
    scanf("%d",&t);
    while(t--){
        scanf("%d",&n);
        for(i=1;i<=n;i++)
            scanf("%d",&r[i]);
        make(1,n,1);
        printf("Case %d:\n",++m);
        while(1){
            scanf("%s",s);
            if(s[0]=='E') break;
            if(s[0]=='Q'){
                scanf("%d %d",&a,&b);
                SUM=0;
                query(a,b,1);
                printf("%d\n",SUM);
            }
            if(s[0]=='A'){
                scanf("%d %d",&a,&b);
                add(a,b,1);
            }
            if(s[0]=='S'){
                scanf("%d %d",&a,&b);
                sup(a,b,1);
            }
        }
    }
    return 0;
}