[poj 3468] A Simple Problem with Integers（裸线段树）

水题，只要注意结果是long long就行了。

#include<iostream>
using namespace std;

#define maxn 100010
struct node
{
	int l;
	int r;
	long long add;
	long long sum;
}tree[maxn * 4];

long long num[maxn];

void down(int num)
{
	tree[num<<1].sum += tree[num].add*(tree[num<<1].r - tree[num<<1].l + 1);
	tree[num<<1|1].sum += tree[num].add*(tree[num<<1|1].r - tree[num<<1|1].l + 1);
	tree[num<<1].add += tree[num].add;
	tree[num<<1|1].add += tree[num].add;
	tree[num].add = 0; 
}

void build(int l, int r, int root)
{
	tree[root].l = l;
	tree[root].r = r;
	tree[root].add = 0;
	if(l == r)
	{
		tree[root].sum = num[l];
		return;
	}
	int mid = (l + r) >> 1;
	build(l, mid, root<<1);
	build(mid + 1, r, root<<1|1);
	tree[root].sum = tree[root<<1].sum + tree[root<<1|1].sum;
}

void update(int l, int r, int num, int root)
{
	if(l <= tree[root].l && tree[root].r <= r)
	{
		tree[root].add += num;
		tree[root].sum += num*(tree[root].r - tree[root].l + 1);
		return;
	}
	if(tree[root].add)
		down(root);
	int mid = (tree[root].l + tree[root].r) >> 1;
	if(l <= mid)
		update(l, r, num, root<<1);
	if(r > mid)
		update(l, r, num, root<<1|1);
	tree[root].sum = tree[root<<1].sum + tree[root<<1|1].sum;
} 

long long query(int l, int r, int root)
{
	if(l <= tree[root].l && tree[root].r <= r)
	{
		return tree[root].sum;
	}
	if(tree[root].add)
		down(root);
	int mid = (tree[root].l + tree[root].r) >> 1;
	long long ans1 = 0, ans2 = 0;
	if(l <= mid)
		ans1 = query(l, r, root<<1);
	if(r > mid)
		ans2 = query(l, r, root<<1|1);
	return ans1 + ans2;
}

int main()
{
	int n, q;
	char ch;
	int a, b, c;
	while(~scanf("%d%d", &n, &q))
	{
		for(int i = 1; i <= n; i++)
			scanf("%lld", &num[i]);
		build(1, n, 1);
		while(q--)
		{
			getchar(); 
			scanf("%c%d%d", &ch, &a, &b);
			if(ch == 'C')
			{
				scanf("%d", &c);
				update(a, b, c, 1);
			}
			else
			{
				printf("%lld\n", query(a, b, 1));
			}
		}
	}
	return 0;
}