POJ 3468 线段树

模板题目。
用线段树就好了。关于线段树参考白皮书。
这里用每一段线段树来更新累加和。注意要用longlong才行。
详细看代码注释。

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<cstdlib>
#include<climits>
#include<stack>
#include<vector>
#include<queue>
#include<set>
#include<map>
#define up(i,a,b)  for(int i=a;i<b;i++)
#define dw(i,a,b)  for(int i=a;i>b;i--)
#define upd(i,a,b) for(int i=a;i<=b;i++)
#define dwd(i,a,b) for(int i=a;i>=b;i--)
//#define local
typedef long long ll;
const double esp = 1e-6;
const double pi = acos(-1.0);
const long long INF = 0x3f3f3f3f;
using namespace std;
typedef pair<int, int> pir;
int n, q;
//ll bit[4 * 100004];
ll add[4 * 100004];//节点的值
//ll sum[4 * 100004];
ll lazy[4 * 100004];//这里用来储存单个加上的值，所以后面是要乘以区间长度的
ll sum = 0;
void pushup(int rt)//回溯父亲节点
{
	add[rt] = add[rt << 1] + add[rt << 1 | 1];
}
void pushdown(int l, int r, int rt)//向子节点推进
{
	int mid = (l + r) >> 1;
	if (lazy[rt])//如果之前这段被更新过了，因为我们只是跟新了线段树中父亲节点来代表之一段更新了
	//那么父亲节点下面的子节点其实还是原来的值，这里用懒惰处理
	{
		add[rt << 1] += (mid - l + 1)* lazy[rt];
		add[rt << 1 | 1] += (r - mid)*lazy[rt];
		lazy[rt << 1] += lazy[rt];//注意要推一下lazy的值，因为可能子节点都还不是叶子
		lazy[rt << 1 | 1] += lazy[rt];
		lazy[rt] = 0;//清楚标记
	}
}
void build(int l,int r,int rt)
{
	lazy[rt] = 0;
	if (l == r)
	{
		scanf("%I64d", &add[rt]);//输入建树
		return;
	}
	int mid = (l + r) >> 1;
	build(l, mid, rt << 1);//先递归左节点
	build(mid + 1, r, rt << 1 | 1);
	pushup(rt);
}
void pluss(int a, int b,int c ,int l, int r, int rt)//这里就是处理给区间加上一个数字
{
	if (a == l && b == r)
	{
		lazy[rt] += (ll)c;
		add[rt] += (ll)(r-l+1)*c;
		return;
	}
	pushdown(l, r, rt);//向下推，一定要向下推，比如更新了1-5，然后更新了节点4，然后查询4 -5，
	//就发现4-5的值是错误的如果不推的话
	int mid = (l + r) >> 1;
	if (b <= mid) pluss(a, b, c, l, mid, rt << 1);//分区间可能进行递推
	else if (a > mid)pluss(a, b, c, mid + 1, r,rt << 1 | 1);
	else {
		pluss(a, mid, c, l, mid, rt << 1);//mid在a，b之间的时候，是不同的，
		//需要更新a，b的值，要不然最后是找不到需要的区间的
		pluss(mid+1, b, c, mid + 1, r, rt << 1 | 1);
	}
	pushup(rt);//回溯向上，因为可能区间并不连续，是几段的和
}

void getsum(int a,int b,int l,int r,int rt)//查询sum和，为了方便直接宏定义sum就行了，然后输出sum
{
	if (a == l && b == r)
	{
		sum += add[rt];
		return;
	}//区间匹配
	pushdown(l, r, rt);
	int mid = (l + r) >> 1;
	if (b <= mid)getsum(a, b, l, mid, rt << 1);//这里递推和上面一样，分了好几种情况，画图就行
	else if (a > mid)getsum(a, b, mid + 1, r, rt << 1|1);
	else {
		getsum(a, mid, l, mid, rt << 1);
		getsum(mid + 1, b, mid + 1, r, rt << 1 | 1);
	}
}
int main()
{
	cin >> n >> q;
	build(1, n, 1);
	char s[2];//用数组取读取第一个字符，不用考虑getchar取提取回车字符，代码要舒服一些
	int a, b, c;
	while (q--)
	{
		scanf("%s", &s);
		if (s[0] == 'Q') {
			cin >> a >> b;
			getsum(a, b, 1, n, 1);
			cout << sum << endl;
			sum = 0;
		}
		else if (s[0] == 'C')
		{
			scanf("%d %d %d", &a, &b, &c);
			pluss(a, b, c, 1, n, 1);
			
		}
	}
//	cout << "**";

	return 0;
}