POJ 3468 A Simple Problem with Integers （线段树 区间修改 区间查询）

POJ 3468 A Simple Problem with Integers 

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

"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

线段树 区间修改 区间查询 。。。存个模板

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#define maxn 100005
#define ll long long
using namespace std;
int num[maxn<<2];
struct node {
	int l,r;
	ll nsum;
	ll add;
} seg[400000];
void Build(int i,int l,int r) {
	seg[i].l=l;
	seg[i].r=r;
	seg[i].add=0;
	if(l==r) {
		seg[i].nsum=num[l];
		return ;
	}
	int mid=l+((r-l)>>1);
	Build(i<<1,l,mid);
	Build(i<<1|1,mid+1,r);
	seg[i].nsum=seg[i<<1].nsum+seg[i<<1|1].nsum;
}
void Add(int i,int l,int r,ll c) {
	if(seg[i].l==l&&seg[i].r==r) {
		seg[i].add+=c;
		return ;
	}
	seg[i].nsum+=c*(r-l+1);
	int mid=(seg[i].l+seg[i].r)>>1;
	if(r<=mid) Add(i<<1,l,r,c);
	else if(l>mid) Add(i<<1|1,l,r,c);
	else {
		Add(i<<1,l,mid,c);
		Add(i<<1|1,mid+1,r,c);
	}
}
ll Query(int i,int a,int b) { //查询a-b的总和
	if(seg[i].l==a&&seg[i].r==b) {
		return seg[i].nsum+(b-a+1)*seg[i].add;
	}
	seg[i].nsum+=(seg[i].r-seg[i].l+1)*seg[i].add;
	int mid=(seg[i].l+seg[i].r)>>1;
	Add(i<<1,seg[i].l,mid,seg[i].add);
	Add(i<<1|1,mid+1,seg[i].r,seg[i].add);
	seg[i].add=0;
	if(b<=mid)  return Query(i<<1,a,b);
	else if(a>mid)  return Query(i<<1|1,a,b);
	else return Query(i<<1,a,mid)+Query(i<<1|1,mid+1,b);
}
int main() {
	int n,m,i;
	int a,b,c;
	char s[10];
	scanf("%d%d",&n,&m);
	for(i=1; i<=n; i++) {
		scanf("%d",&num[i]);
	}
	Build(1,1,n);
	for(i=0; i<m; i++) {
		scanf("%s",s);
		if(s[0]=='C') {
			scanf("%d%d%d",&a,&b,&c);
			Add(1,a,b,c);
		} else {
			scanf("%d%d",&a,&b);
			printf("%lld\n",Query(1,a,b));
		}

	}
	return 0;
}