用線段樹模板(ACM典型線段樹)

題目   敵兵佈陣 node

Problem Description 算法

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

Input code

第一行一個整數T,表示有T組數據。
每組數據第一行一個正整數NN<=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條命令 ip

 Output io

對第i組數據,首先輸出「Case i:」和回車,
對於每一個Query詢問,輸出一個整數並回車,表示詢問的段中的總人數,這個數最多不超過1000000 模板

Sample Input class

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

題目分析:用線段樹模板

#include <stdio.h>

#define M 50005 /*空間要足夠大*/


struct node

{

    int l,r,c;

};


int total,times,data[M];

struct node arr[M*3];  /*樹節點*/


int Construct(int l,int r,int k)  /*創建線段樹*/

{

    int mid;

    if(l==r)

    {

        arr[k].l=arr[k].r=r;

        return arr[k].c=data[l];

    }

    mid=(l+r)/2;

    arr[k].l=l;

    arr[k].r=r;

    return (arr[k].c=Construct(l,mid,2*k)+Construct(mid+1,r,2*k+1));

}



void Deal(int l,int r,int k,int loc,int num)   /*當前節點loc權值加num 加的時候num爲正值,減的時侯num爲負數*/

{

    int mid;

    if(arr[k].l==arr[k].r&&arr[k].l==loc)   /*找到要更新的節點*/

    

    {

        arr[k].c=arr[k].c+num;    /*權值加num*/

        return;

    }

     mid=(l+r)/2;

    

    arr[k].c+=num;  /*k不是所要找的節點,那麼要找的應該在它的子樹上,因此區間權值也要加上num*/

    

    

    if(loc<=mid)  /*所要找的節點,在左邊*/

    {

        Deal(l,mid,2*k,loc,num);

    }

    else

    {

        Deal(mid+1,r,2*k+1,loc,num);


    }

}



int Query(int l,int r,int k)

{

    int mid;

    if(arr[k].l==l&&arr[k].r==r)

    {

        return arr[k].c;

    }

    mid=(arr[k].l+arr[k].r)/2;

    if(r<=mid)

    {

        return Query(l,r,2*k);


    }

    else if(l>mid)

    {

        return Query(l,r,2*k+1);

    }

    else

    {

        return Query(l,mid,2*k)+Query(mid+1,r,2*k+1);

    }


}




int main()

{

    int m,n,i, count=0;

    scanf("%d",×);

    while(times--)

    {

        printf("Case %d:\n",++count);

        scanf("%d",&total);

        for(i=1;i<=total;i++)

        {

            scanf("%d",&data[i]);

        }

        Construct(1,total,1);

        while(1)

        {

            char temp[10];

            scanf("%s",temp);

            if(temp[0]=='E')

                break;

        

            scanf("%d%d",&m,&n);

            if(temp[0]=='Q')

            {

                printf("%d\n",Query(m,n,1));


            }

            else if(temp[0]=='A')

                Deal(1,total,1,m,n);

            else if(temp[0]=='S')

                Deal(1,total,1,m,-n);

        }

    }

return 0;

}
相關文章
相關標籤/搜索