HDU 1166 敌兵布阵(线段树）

题目地址：HDU 1166

听说胡浩版的线段树挺有名的。于是就拜访了一下他的博客。详情戳这里。于是就完全仿照着胡浩大牛的风格写的代码。

至于原理，鹏鹏学长已经讲的再清晰不过了。我就在下面的代码注释中将原理说明一下吧。来纪念第一发线段树。

下面是代码+注释。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <math.h>
#include <stack>
using namespace std;
#define lson l, mid, rt<<1//直接定义子节点，因为每次都要用到，所以直接定义一个很方便
#define rson mid+1, r, rt<<1 | 1
const int MAXN=51000;
int sum[MAXN<<2];
void PushUP(int rt)//向上更新父节点的值
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void build(int l, int r, int rt)//建立二叉树
{
    if(l==r)//已经到达最底端的叶子节点，即单点，直接输入该值
    {
        scanf("%d",&sum[rt]);
        return ;
    }
    int mid=(l+r)>>1;
    build(lson);//向左子节点继续建立二叉树
    build(rson);//向右子节点继续建立二叉树
    PushUP(rt);//全部建立完后向上更新父节点的值
}
void update(int p, int x, int l, int r, int rt)//单点修改
{
    if(l==r)//说明已经到了最底端的叶子节点，已经是单点了，可以直接修改该值
    {
        sum[rt]+=x;
        return ;
    }
    int mid=(l+r)>>1;
    if(p<=mid) update(p,x,lson);//如果要修改的值在这个区间左边，就进入左子节点继续寻找
    else update(p,x,rson);//如果要修改的值在这个区间右边，就进入右子节点继续寻找
    PushUP(rt);//修改完后，仍然要向上更新父节点的值
}
int query(int ll, int rr, int l, int r, int rt)//区间查询
{
    if(ll<=l&&rr>=r)//如果要查询的区间完全覆盖了该子节点，直接返回该子节点的值
        return sum[rt];
    int mid=(l+r)>>1;
    int ans=0;
    if(ll<=mid) ans+=query(ll,rr,lson);//如果在该子节点左边还有一部分要查询的区间，就去左子树继续查询
    if(rr>mid) ans+=query(ll,rr,rson);//如果在该子节点右边还有一部分要查询的区间，就去右子树继续查询
    return ans;
}
int main()
{
    int t, n, i, a, b, ans, num=0;
    char s[20];
    scanf("%d",&t);
    while(t--)
    {
        num++;
        printf("Case %d:\n",num);
        memset(sum,0,sizeof(sum));
        scanf("%d",&n);
        build(1,n,1);//建立
        /*for(i=1;i<=25;i++)
        {
            printf("%d    %d\n",i,sum[i]);
        }*/
        getchar();
        while(scanf("%s",s))
        {
            if(s[0]=='E') break;
            if(!strcmp(s,"Add"))
            {
                scanf("%d%d",&a,&b);
                update(a,b,1,n,1);//修改
            }
            else if(!strcmp(s,"Sub"))
            {
                scanf("%d%d",&a,&b);
                update(a,-b,1,n,1); //修改
            }
            else
            {
                scanf("%d%d",&a,&b);
                ans=query(a,b,1,n,1);//查询
                printf("%d\n",ans);
            }
        }
    }
    return 0;
}