POJ-3468

成段更新的线段树,应该也算比较基础的写法

#include<cstdio>
#include<algorithm>
#include<cstring>

using namespace std;

namespace
{
	typedef long long ll;
	const int MAX_N = (1 << 18) - 1;
	int n, N, Q, a, b, c;
	ll da[MAX_N], db[MAX_N];

	void add(int k, int l, int r)
	{
		if (a <= l && r <= b)
			da[k] += c;
		else if (!(b <= l || a >= r))
		{
			db[k] += (min(r, b) - max(l, a)) * c;
			add(k * 2 + 1, l, (l + r) / 2);
			add(k * 2 + 2, (l + r) / 2, r);
		}
	}

	ll sum(int k, int l, int r)
	{
		if (b <= l || a >= r)
			return 0;
		else if (a <= l && r <= b)
			return (r - l) * da[k] + db[k];
		else
		{
			ll res = (min(r, b) - max(l, a)) * da[k];
			res += sum(k * 2 + 1, l, (l + r) / 2);
			res += sum(k * 2 + 2, (l + r) / 2, r);
			return res;
		}
	}

	ll init(int k)
	{
		if (k >= n - 1)
			return db[k];
		else
			return db[k] = init(2 * k + 1) + init(2 * k + 2);
	}

	void input()
	{
		memset(da, 0, sizeof(da));
		memset(db, 0, sizeof(db));
		scanf("%d %d", &N, &Q);
		n = 1;
		while (n < N)
			n <<= 1;
		for (int i = 0; i < N; i++)
			scanf("%lld", &db[i + n - 1]);
		init(0);
		char s[2];
		while (Q--)
		{
			scanf("%s", s);
			if (s[0] == 'Q')
			{
				scanf("%d %d", &a, &b);
				a--;
				printf("%lld\n", sum(0, 0, n));
			}
			else
			{
				scanf("%d %d %d", &a, &b, &c);
				a--;
				add(0, 0, n);
			}
		}
	}

}

int main()
{
	input();
	return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值