线段树单点更新hdu1166

学习自https://blog.csdn.net/metalseed/article/details/8039326#t3

先序建子树，然后回溯建立父节点

 

更新时

先递归更新子节点，然后再回溯更新父节点

 回溯思想很重要

#include <cstdio>
using namespace std;
const int maxn = 50005;
int sum[maxn<<2];
void build(int l, int r, int rt);
void pushup(int rt);
int query(int L, int R, int l, int r, int rt);
void update(int p, int add, int l, int r, int rt);
void update(int p, int add, int l, int r, int rt);
int main()
{
    int i;
    int t, n;
    scanf("%d", &t);
    for(i = 1; i <= t; ++i)
    {
        printf("Case %d:\n", i);
        scanf("%d", &n);
        build(1, n, 1);
        char op[10];
        while(scanf("%s", op) && op[0] != 'E')
        {
            int a, b;
            scanf("%d %d", &a, &b);
            if(op[0] == 'Q')
            {
                int ans = query(a, b, 1, n, 1);
                printf("%d\n", ans);
            }
            else if(op[0] == 'S') update(a, -b, 1, n, 1);
            else update(a, b, 1, n, 1);
        }
    }
    return 0;
}
void update(int p, int add, int l, int r, int rt)
{
    if(l == r)
    {
        sum[rt] += add;
        return;
    }
    int mid = (l + r)>>1;
    if(p <= mid) update(p, add, l, mid, rt<<1);              //若p在左子树里，就只更新左子树
    else update(p, add, mid + 1, r, rt<<1|1);                //若p在右子树里，就只更新右子树
    pushup(rt);
}
int query(int L, int R, int l, int r, int rt)                //前两个变量大写const值，表示查询的区间；后三个变量绑定，表示当前节点的区间和序号
{
    if(L <= l && r <= R) return sum[rt];                     //当到达该状态时，即当前节点的区间全在查询区间内，就停止向下搜素
    int ans = 0;
    int mid = (l + r)>>1;                                    //该节点的左子节点的右边界，右子节点的左边界减1
    if(L <= mid) ans += query(L, R, l, mid, rt<<1);          //若该节点的左子节点的右边界不小于查询区间的左边界，则有往下搜索的价值
    if(R >= mid + 1) ans += query(L, R, mid + 1, r, rt<<1|1);//若右子节点的左边界不大于查询区间的右边界，则有往下搜索的价值
    return ans;
}
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(l, mid, rt<<1);
    build(mid + 1, r, rt<<1|1);
    pushup(rt);                                             //回溯更新父节点
}