POJ 3468 A Simple Problem with Integers（线段树、lazy思想）

题目大意是给出一串数学，之后可能对某个区间内的所有数字都增加一定的量，也可能对某个区间进行查询，要求返回这个区间内所有数字的和。

Training 给出了8 道关于线段树的题目，手气好第一道就抽中最简单的。但是还是被折腾了不少时间，一开始用最简单的方法对每个区间进行update，试了几次都是TLE。后来才发现其实可以不用每次都update所有区间。Update时，如果到一整个区间都需要update的 节点，可以先将其标记，不对其下面的节点进行处理。当下次要查询，或者更新这个节点下面的节点时，才进行下面节点的更新。这样可以大大节省时间。

#include<iostream>
#include<cstdio>

#define MID(x, y) (((x) + (y)) >> 1)
#define L(x) (x << 1)
#define R(x) (x << 1 | 1)
using namespace std;

typedef struct{
	int l, r;
	long long sum;
	long long offset;
} NODE;

NODE st[440000];
int input[110000];

void build(int t, int l, int r){
	st[t].l = l;
	st[t].r = r;
	st[t].offset = 0;
	if(l == r){
		st[t].sum = input[l];
	}else{
		build(L(t), l, MID(l, r));
		build(R(t), MID(l, r) + 1, r);
		st[t].sum = st[L(t)].sum + st[R(t)].sum;
	}
}

void add(int t, int l, int r, int a){
	int mid = MID(st[t].l, st[t].r);
	st[t].sum += a * (r - l + 1);
	if(st[t].l == l && st[t].r == r){
		st[t].offset += a;
		return;
	}
	if(st[t].offset){
		st[R(t)].offset += st[t].offset;
		st[L(t)].offset += st[t].offset;
		st[L(t)].sum += st[t].offset * (st[L(t)].r - st[L(t)].l + 1);
		st[R(t)].sum += st[t].offset * (st[R(t)].r - st[R(t)].l + 1);
		st[t].offset = 0;
	}
	if(mid >= r){
		add(L(t), l, r, a);
	}else if(mid < l){
		add(R(t), l, r, a);
	}else{
		add(L(t), l, mid, a);
		add(R(t), mid + 1, r, a);
	}
}

long long query(int t, int l, int r){
	int mid = MID(st[t].l, st[t].r);
//	cout << "mid: " << mid << endl;
	if(st[t].l == l && st[t].r == r){
		return st[t].sum; 
	}
	if(st[t].offset){
		st[R(t)].offset += st[t].offset;
		st[L(t)].offset += st[t].offset;
		st[L(t)].sum += st[t].offset * (st[L(t)].r - st[L(t)].l + 1);
		st[R(t)].sum += st[t].offset * (st[R(t)].r - st[R(t)].l + 1);
		st[t].offset = 0;
	}
	if(mid >= r){
		return query(L(t), l, r);
	}else if(mid < l){
		return query(R(t), l, r);
	}else{
		return query(L(t), l, mid) + query(R(t), mid + 1, r);
	}
}

int main(){
	int N, Q;
	scanf("%d %d", &N, &Q);
	for(int i = 1; i <= N; i++)
		scanf("%d", &input[i]);
	build(1, 1, N);
	while(Q--){
		char cmd;
		scanf("%c", &cmd);
		scanf("%c", &cmd);
		if(cmd == 'Q'){
			int l, r;
			scanf("%d %d", &l, &r);
			printf("%I64d\n", query(1, l, r));
		}else{
			int l, r;
		       	long long a;
			scanf("%d %d %I64d", &l, &r, &a);
			add(1, l, r, a);
		}
	}
}