poj 3468(线段树)

最新推荐文章于 2018-04-20 20:32:48 发布

原创最新推荐文章于 2018-04-20 20:32:48 发布 · 483 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#数据结构 #线段树

ACM-线段树专栏收录该内容

54 篇文章

订阅专栏

本文介绍了一种基于线段树的数据结构解决区间修改和查询的问题。通过递归建立、更新和查询线段树来高效地处理区间加法操作及求区间和。文章提供了完整的C++实现代码。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意：一个数字序列，Q a b查询a到b的数字和，C a b c从a到b都加c。

题解：区间修改水题。

#include <stdio.h>
#include <string.h>
const int N = 100005;
long long addv[N << 2], sum[N << 2], x;
int n, q, l1, r1;

void pushup(int k, int left, int right) {
	sum[k] = sum[k * 2] + sum[k * 2 + 1];
}

void pushdown(int k, int left, int right) {
	int len = right - left + 1;
	if (addv[k]) {
		addv[k * 2] += addv[k];
		addv[k * 2 + 1] += addv[k];
		sum[k * 2] += addv[k] * (len - len / 2);
		sum[k * 2 + 1] += addv[k] * (len / 2);
		addv[k] = 0;
	}
}

void build(int k, int left, int right) {
	addv[k] = 0;
	if (left == right) {
		scanf("%lld", &sum[k]);
		return;
	}
	int mid = (left + right) / 2;
	build(k * 2, left, mid);
	build(k * 2 + 1, mid + 1, right);
	pushup(k, left, right);
}

void update(int k, int left, int right) {
	if (l1 <= left && right <= r1) {
		addv[k] += x;
		sum[k] += x * (right - left + 1);
		return;
	}
	pushdown(k, left, right);
	int mid = (left + right) / 2;
	if (l1 <= mid)
		update(k * 2, left, mid);
	if (r1 > mid)
		update(k * 2 + 1, mid + 1, right);
	pushup(k, left, right);
}

long long query(int k, int left, int right) {
	if (l1 <= left && right <= r1)
		return sum[k];
	pushdown(k, left, right);
	int mid = (left + right) / 2;
	long long res = 0;
	if (l1 <= mid)
		res += query(k * 2, left, mid);
	if (r1 > mid)
		res += query(k * 2 + 1, mid + 1, right);
	return res;
}

int main() {
	while (scanf("%d%d", &n, &q) == 2) {
		build(1, 1, n);
		char str[5];
		while (q--) {
			scanf("%s%d%d", str, &l1, &r1);
			if (str[0] == 'Q')
				printf("%lld\n", query(1, 1, n));
			else if (str[0] == 'C') {
				scanf("%lld", &x);
				getchar();
				update(1, 1, n);
			}
		}
	}
	return 0;
}