HDU 1166 敌兵布阵 // 线段树

题目描述

HDU 1166 敌兵布阵

解题思路

线段树的入门基础题.
题目要求支持下列操作:
1) 区间查询 (求和)
2) 单点更新 (加\减)
// 具体还是看注释哈~^ ^

参考代码

#include <stdio.h>#define lson rt<<1  // 左儿子的下标#define rson rt<<1|1    // 右儿子的下标#define mid ((l + r) >> 1)  // 区间[l,r]的中点const int MAX_N = 50010;int sum[MAX_N << 2]; // 一般开最大数据量的4倍 // 把当前节点的信息更新到父节点(root, 简称 rt)void pushup(int rt){        sum[rt] = sum[lson] + sum[rson];}// 对以rt为根节点的区间[l, r]建树void build(int l, int r, int rt){       if (l == r){ // 到达叶节点         scanf("%d", &sum[rt]);        return ;    }    build(l, mid, lson); // 左儿子区间    build(mid+1, r, rson); // 右儿子区间    pushup(rt); // 更新当前节点的信息}// 对第k个点 ( k∈[1, n] ) 进行更新(加上num) void update(int k, int num, int l, int r, int rt){      if (l == r){ // 到达叶节点        sum[rt] += num;        return ;    }    if (k <= mid)   update(k, num, l, mid, lson); // 如果k在rt的左儿子区间里, 则更新左区间    else    update(k, num, mid+1, r, rson); // 否则 更新右区间    pushup(rt); }// 从以rt为根节点的区间[l, r] 中 查询 区间[L, R]int query(int L, int R, int l, int r, int rt){    if (L <= l && R >= r){ // 如果待查询区间[L, R] 完全包含 区间[l, r] 则 返回区间[l, r]所维护的信息        return sum[rt];    }    int ans = 0;        if (L <= mid)   ans += query(L, R, l, mid, lson); // 区间[l, r] 的左儿子区间包含[L, R]的一部分 则 查询左区间    if (R > mid)    ans += query(L, R, mid+1, r, rson); // 区间[l, r] 的右儿子区间包含[L, R]的一部分 则 查询右区间    return ans;}int main(){    int T, n, a, b, kase = 1;    char opr[10];    scanf("%d", &T);    while (T--){        scanf("%d", &n);        build(1, n, 1);        printf("Case %d:\n", kase++);        while (scanf("%s", opr) && opr[0] != 'E'){            scanf("%d %d", &a, &b);            if (opr[0] == 'Q'){                printf("%d\n", query(a, b, 1, n, 1));            }else{                b = (opr[0] == 'A' ? b : -b);                update(a, b, 1, n ,1);            }        }    }    return 0;}