线段树模板-单点增减区间求和模板

最新推荐文章于 2019-02-18 16:23:48 发布

原创最新推荐文章于 2019-02-18 16:23:48 发布 · 668 阅读

1 ·

CC 4.0 BY-SA版权

线段树（模板来自胡浩大神）专栏收录该内容

2 篇文章

订阅专栏

本文介绍了一种使用段式树状数组实现的数据结构，该结构支持快速查询区间和及更新单个元素值的操作。文章详细展示了如何通过递归方式构建、更新数据以及查询指定区间的元素总和，并提供了一个完整的C语言程序示例。

#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];
}
void build(int l,int r,int rt) {
	if (l == r) {
		scanf("%d",&sum[rt]);
		return ;
	}
	int m = (l + r) >> 1;
	build(lson);
	build(rson);
	PushUP(rt);
}
void update(int p,int add,int l,int r,int rt) {  //  第 p项 加 add   
	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 L,int R,int l,int r,int rt) {   //  求区间 L 到 R的和
	if (L <= l && r <= R) {
		return sum[rt];
	}
	int m = (l + r) >> 1;
	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);
		build(1 , n , 1);  // 建树 1 - n
		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));  // a到 b的 和
			else if (op[0] == 'S') update(a , -b , 1 , n , 1);          // 第 a项 减 b
			else update(a , b , 1 , n , 1);                             // 第 a项 加 b
		}
	}
	return 0;
}