线段树模版—A Simple Problem with Integers

D - A Simple Problem with Integers

Time Limit:5000MS     Memory Limit:131072KB     64bit IO Format:%I64d & %I64u

Description

给出了一个序列，你需要处理如下两种询问。

"C a b c"表示给[a, b]区间中的值全部增加c (-10000 ≤ c ≤ 10000)。

"Q a b" 询问[a, b]区间中所有值的和。

Input

第一行包含两个整数N, Q。1 ≤ N,Q ≤ 100000.

第二行包含n个整数，表示初始的序列A (-1000000000 ≤ Ai ≤ 1000000000)。

接下来Q行询问，格式如题目描述。

Output

对于每一个Q开头的询问，你需要输出相应的答案，每个答案一行。

Sample Input

10 5
1 2 3 4 5 6 7 8 9 10
Q 4 4
Q 1 10
Q 2 4
C 3 6 3
Q 2 4

Sample Output

4
55
9
15

这题目。。。难吗？真不难，就是一个完完整整的模板题，但是我错了四次，没找出自己哪里错了。。。惭愧啊，虽然我一直盯着A (-1000000000 ≤ Ai ≤ 1000000000)

我竟然蠢到只是在判断他是不是整数，而忘记了我们更新操作中，加上一个数可能让它爆了整数范围，更重要的是求和！我连求和都没有考虑进来啊！！几个差不多爆掉整形的数字相加肯定爆了范围了，我却还是无动于衷。。。天啊，太笨了啊，参加acm的意义莫过于发现自己到底有多蠢，同样年龄的那些人多么的聪明。

所以，把query返回值改为long long，结构体中的sum和lazy改为long long，或者你全部改成long long。。。就这样，ac了。

啊，不是方法的问题，而是数据上处理的问题，死在这地方真丢人啊。包括以前没有打表导致一直TLE。。。思维的严谨是一个人最好的优点了吧。

代码如下：

#include
#include
#include
using namespace std;
const int Maxn = 100000 + 5;
int a[Maxn],val;
void pushdown(int node);

struct seg
{
	long long sum,lazy;
	int l,r;
}segtree[Maxn*4];

void build(int node,int l,int r)
{
    segtree[node].l=l;
	segtree[node].r=r;
	segtree[node].lazy=0;
    if(l==r)
    {
    	segtree[node].sum=a[l];
		return;
	}
	else
	{
		int mid=(l+r)>>1;
	    build(node<<1,l,mid);
	    build(node<<1|1,mid+1,r);
	    segtree[node].sum = segtree[node<<1].sum + segtree[node<<1|1].sum;
	}
}

long long query(int node,int x,int y)
{
	if(segtree[node].l == x && segtree[node].r == y)
	{
		return segtree[node].sum;
	}
	pushdown(node);
	int mid = segtree[node].l + segtree[node].r;
	long long res = 0;
	mid >>= 1;
	if(mid >= y)
		res += query(node<<1,x,y);
	else if(mid < x)
		res += query(node<<1|1,x,y);
	else
	{
		res += query(node<<1,x,mid);
		res += query(node<<1|1,mid+1,y);
	}
	return res;
}

void pushdown(int node)
{
	if(segtree[node].lazy != 0)
	{
		segtree[node<<1].lazy += segtree[node].lazy;
		segtree[node<<1|1].lazy += segtree[node].lazy;
		segtree[node<<1].sum += segtree[node].lazy * (segtree[node<<1].r - segtree[node<<1].l + 1);
		segtree[node<<1|1].sum += segtree[node].lazy * (segtree[node<<1|1].r - segtree[node<<1|1].l + 1);;
		segtree[node].lazy = 0;
	}
}

void update(int node,int x,int y)
{
	if(segtree[node].l == x && segtree[node].r == y)
	{
		segtree[node].lazy += val;
		segtree[node].sum += val * (y - x + 1);
		return;
	}
	if(segtree[node].l == segtree[node].r)
		return;
	pushdown(node);
	int mid = segtree[node].l + segtree[node].r;
	mid >>= 1;
	if(mid >= y)
		update(node<<1,x,y);
	else if(mid < x)
		update(node<<1|1,x,y);
	else
	{
		update(node<<1,x,mid);
		update(node<<1|1,mid+1,y);
	}
	segtree[node].sum = segtree[node<<1].sum + segtree[node<<1|1].sum;
}

int main()
{
	int t,n,q,l,r,i,k=1;
	char ch[2];
	while(scanf("%d%d",&n,&q)!=EOF) 
	{
		for(i=1;i<=n;i++)
			scanf("%d",&a[i]);
		build(1,1,n);
		while(q--)
		{
			scanf("%s",&ch);
			if(ch[0]=='Q')
			{
				scanf("%d%d",&l,&r);
				printf("%lld\n",query(1,l,r));
			}
			else
			{
				scanf("%d%d%d",&l,&r,&val);
				update(1,l,r);
			}
		}
	}
	return 0;
}