线段树-单点更新模版

/**
*模版原题:HDU 1754
*线段树-单点加减
*区间求最值
*/

#include <cstdio> #define lson l , m , rt << 1 #define rson m + 1 , r , rt << 1 | 1 const int maxn = 55555; int sum[maxn<<2]; void PushUP(int rt) { sum[rt] = sum[rt<<1] + sum[rt<<1|1]; // sum[rt]=sum[rt*2]+sum[rt*2+1]; } //建立线段树; //l,r为该点包含的范围 rt为该点所在数组的下标 void build(int l,int r,int rt) { if (l == r) //当查询到的点为最终的单个点时; { scanf("%d",&sum[rt]); return ; } int m = (l + r) >> 1; //m=(l+r)/2; build(lson); build(rson); PushUP(rt); //回溯更新父节点 } //单点修改(加) //p为待更新点下标 add为待更新参数值 l,r,rt为查找范围和范围内的最高祖先 void update(int p,int add,int l,int r,int rt) { if (l == r) //如果查找到该点为最终单点,执行更新 { sum[rt] += add; return ; } int m = (l + r) >> 1; //如非最终点,取范围中间值判断目标点是在该点左子树还是右子树 if (p <= m) update(p , add , lson); else update(p , add , rson); PushUP(rt); //回溯更新父节点 } //查询线段树 //L R为输入的区间 l r为查找范围和范围内的最高祖先 int query(int L,int R,int l,int r,int rt) { if (L <= l && r <= R) //如果预先出现某节点范围属于原始L R内,赋予该节点值。不再查询该节点的子节点 { return sum[rt]; } int m = (l + r) >> 1; //如为如上退出,这判断L R 包含的区域是在该点左右哪子树 int ret = 0; if (L <= m) ret += query(L , R , lson); if (R > m) ret += query(L , R , rson); return ret; } int main() { int T , n; scanf("%d",&T); for (int cas = 1 ; cas <= T ; cas ++) { printf("Case %d:\n",cas); scanf("%d",&n); //线段树起点下标由1开始到n build(1 , n , 1); char op[10]; while (scanf("%s",op)) { if (op[0] == 'E') break; int a , b; scanf("%d%d",&a,&b); if (op[0] == 'Q') printf("%d\n",query(a , b , 1 , n , 1)); else if (op[0] == 'S') update(a , -b , 1 , n , 1); else update(a , b , 1 , n , 1); } } return 0; }

 

posted @ 2016-01-25 10:34  蓦辰  阅读(117)  评论(0编辑  收藏  举报