1. 程式人生 > >hdu 1166 敵兵佈陣(線段樹點區)

hdu 1166 敵兵佈陣(線段樹點區)

題目大意:   給出初始化的區間值,然後有三種詢問

                    Query a b 詢問區間[a,b]值的總和

                    Add a b 第a個元素的值加b

                    Sub a b 第a個元素的值減b

解題思路:  線段樹 更新:單點增減 詢問:區間和

                   每次更新在結點儲存左右子樹值的和,查詢時就不需要查到最低,實現區間查詢

                   更新時間複雜度O(logN),查詢時間複雜度O(logN)

程式碼:

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#define MAX 100000
#define MID(a,b) (a+b)>>1
#define L(a) a<<1
#define R(a) (a<<1)+1
typedef struct snode{
    int num,left,right;
}Node;
int num[MAX];
Node Tree[MAX<<1];

void Init()
{
    memset(Tree,0,sizeof(Tree));
}

void Build(int t,int l,int r)    //以t為根結點,建立左子樹為l,右子樹為r的線段樹
{
    int mid;
    Tree[t].left=l,Tree[t].right=r;
    if(Tree[t].left==Tree[t].right)
    {
        Tree[t].num=num[l];
        return ;
    }
    mid=MID(Tree[t].left,Tree[t].right);
    Build(L(t),l,mid);
    Build(R(t),mid+1,r);
    Tree[t].num=Tree[L(t)].num+Tree[R(t)].num;
}

void Insert(int t,int l,int r,int n)  //向t為根結點,左子樹為l,右子樹為r的結點加上值n
{
    int mid;
    if(Tree[t].left==l&&Tree[t].right==r)
    {
        Tree[t].num+=n;
        return ;
    }
    mid=MID(Tree[t].left,Tree[t].right);
    if(r<=mid)
        Insert(L(t),l,r,n);
    else if(l>=mid)
        Insert(R(t),l,r,n);
    else
    {
        Insert(L(t),l,mid,n);
        Insert(R(t),mid+1,r,n);
    }
    Tree[t].num+=n;
}

int Query(int t,int l,int r)    //查詢根結點為t,左子樹為l,右子樹為r的結點的值
{
    int mid;
    if(Tree[t].left==l&&Tree[t].right==r)
    {
        return Tree[t].num;
    }
    mid=MID(Tree[t].left,Tree[t].right);
    if(l>mid)        //***
        return Query(R(t),l,r);
    else if(r<=mid)   //***
        return Query(L(t),l,r);
    else
    {
        int a,b;
        a=Query(L(t),l,mid);
        b=Query(R(t),mid+1,r);
        return a+b;
    }
}

int main()
{
    char ch[10];
    int t,n,i,i1,a,b;
    scanf("%d",&t);
    for(i1=1;i1<=t;i1++)
    {
        Init();         //初始化
        scanf("%d",&n);
        for(i=1;i<=n;i++)
            scanf("%d",&num[i]);
        Build(1,1,n);   //以1為根結點建立線段樹
        printf("Case %d:\n",i1);
        while(scanf("%s",ch)&&strcmp(ch,"End")!=0)
        {
            scanf("%d%d",&a,&b);
            if(strcmp(ch,"Add")==0)     //第a個元素加b
                Insert(1,a,a,b);
            else if(strcmp(ch,"Sub")==0)  //第a個元素減b
                Insert(1,a,a,-b);
            else
                printf("%d\n",Query(1,a,b));  //查詢[a,b]區間值的總和
        }
    }
    return 0;
}
注:原創文章,轉載請註明出處