树线段hdu 1166 敌兵布阵(线段树)

最近研究树线段，稍微总结一下，以后继续补充：

    

    题目链接:  http://acm.hdu.edu.cn/showproblem.php?pid=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;
}

    注:原创文章，转载请注明出处

    

文章结束给大家分享下程序员的一些笑话语录： Borland说我很有前途，Sun笑了；Sun说我很有钱，IBM笑了；IBM说我很专业，Sybase笑了；Sybase说我数据库很牛，Oracle笑了；Oracle说我是开放的，Linux笑了；Linux说我要打败Unix，微软笑了；微软说我的系统很稳定，我们都笑了。

--------------------------------- 原创文章 By
树和线段
---------------------------------