#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#define lson l, m, rt<<1//左子树的rt即当前节点在这个树中的位置(编号)为每次乘2
#define rson m+1, r, rt<<1|1//右子树树的当前节点的编号为乘2加1
using namespace std;
const int maxn = 55555;
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 m;
m = (l+r)>>1;
build(lson);//(define 的作用)递归建树
build(rson);
pushup(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);
}
int query(int newL, int newR, int l, int r, int rt)
{
//cout << 1<< endl;
if(newL <= l && newR >= r) return sum[rt];
int m = (l+r)>>1;
int ret = 0;
if(newL <= m) ret += query(newL, newR, lson);//在左子树上所要求的左区间小于等于m就递归求和
if(newR > m) ret += query(newL, newR, rson);// 在右子树上m大于所求区间的右区间就递归求和
return ret;
}
int main()
{
int T, n;//T多组输入,n总区间的大小
scanf("%d", &T);
for(int cas = 1; cas <= T; cas++){
memset(sum, 0, sizeof(sum));
cout << "Case " << cas << ":" << endl;
scanf("%d", &n);//这里输入总区间大小后就直接建树,简述过程为输入一个值建一个节点所以在函数中第二次输入
build(1, n, 1);//建树左右区间,以及每个节点
while(1){
string s;
cin >> s;
if(s == "End") break;
int a, b;
scanf("%d%d", &a, &b);
if(s == "Query") cout << query(a, b, 1, n, 1) << endl;//给定区间求和
else if(s == "Sub") update(a, -b, 1, n, 1);//单点减区间加一个负数
else update(a, b, 1, n, 1);
}
}
return 0;
}
/*顺便再次膜拜一下HH神不然不会有这么简洁的代码,理解很清楚*/
hdu1166线段树的点跟新模板
最新推荐文章于 2019-08-20 02:18:21 发布