HDU 1166 敵兵布陣 線段樹

標籤：c++

Problem DescriptionC國的死對頭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以內。
 

#include<iostream>#include<cstring>#include<cstdio>using namespace std;#define qq  50005struct node{   int left,right,value;}q[4*qq];                              //根據二叉樹性質   數組大小不能少於節點數int w[qq];void build(int l,int r,int i)          // 建樹{    q[i].left=l;    q[i].right=r;    if(l==r)    {        q[i].value=w[l];                        //最小的節點代表線段(x,x);        return;    }    int m=(l+r)/2;    build(l,m,2*i);                            // 建左子節點和右子節點    build(m+1,r,2*i+1);    q[i].value=q[i*2].value+q[i*2+1].value;     //父節點為兩個子節點之和}void update(int i,int x,int op,int num)        //更新線段樹{    if(q[i].left==x&&q[i].right==x)    {        if(op==1)            q[i].value+=num;        else            q[i].value-=num;        return;    }    int m=(q[i].left+q[i].right)/2;    if(x<=m)                                 //  如果目標點在該節點的左節點        update(2*i,x,op,num);    else        update(2*i+1,x,op,num);    if(op==1)                                // 回溯回來父節點也要更新        q[i].value+=num;    else        q[i].value-=num;}int query(int l,int r,int i)                 // 詢問樹種的值{    if(q[i].left==l&&q[i].right==r)    {        return q[i].value;                        }    int m=(q[i].left+q[i].right)/2;    if(m>=r)                                 //如果要詢問的右邊在該節點的左邊        return query(l,r,2*i);    else if(l>m)        return query(l,r,2*i+1);    else        return query(l,m,2*i)+query(m+1,r,2*i+1);}int main(){    int t,i,j,n,a,b,s;    char ww[20];    cin>>t;    for(i=1;i<=t;i++)    {        printf("Case %d:\n",i);        scanf("%d",&n);        for(j=1;j<=n;j++)            scanf("%d",&w[j]);            build(1,n,1);        while(scanf("%s",ww))        {            if(ww[0]=='E')                break;            scanf("%d%d",&a,&b);            if(ww[0]=='Q')            {                s=query(a,b,1);                cout<<s<<endl;            }            else if(ww[0]=='A')            {                update(1,a,1,b);            }            else            {                update(1,a,2,b);            }        }    }    return 0;}

Sample Input

1101 2 3 4 5 6 7 8 9 10Query 1 3Add 3 6Query 2 7Sub 10 2Add 6 3Query 3 10End 

 
Sample Output

Case 1:63359充分體現了 資料結構 對時間的最佳化線段樹是二叉樹的一種 每個節點都代表一條線段。。代碼：