Tunnel Warfare HDU - 1540

这题思路用了 ACM ! 荣耀之路 ！ 的代码。 当时看了后一脸懵逼， 还是自己太菜， 刚接触线段树。 

题意：D代表破坏村庄，R代表修复最后被破坏的那个村庄，Q代表询问包括x在内的最大连续区间是多少

思路：在线段树的区间内，我们要用三个变量记录左边连续区间，右边连续区间和最大连续区间， 建议手动推一下

#include <bits/stdc++.h>
using namespace std;
const int maxn = 5e4+10;

struct Node {
	int l, r; //左孩子右孩子 
	int ls, rs, ms; // ls为左端最大连续区间 rs为右端最大连续区间 ms为区间内最大连续区间 
}tree[maxn << 2];

int n, m;
stack<int> s; //保存最近破坏的那个村庄 

void init(int l, int r, int rt) {
	while (!s.empty()) s.pop();
	tree[rt].l = l;
	tree[rt].r = r;
	tree[rt].ls = tree[rt].rs = tree[rt].ms = r-l+1; //将最大区间都置为区间长度 
	if (l != r) {
		int mid = (l+r) >> 1;
		init(l, mid, rt<<1);
		init(mid+1, r, rt<<1|1);
	}
}

void insert(int rt, int t, int flag) { // flag为破坏或者修复 
	if (tree[rt].l == tree[rt].r) {
		if (flag)
			tree[rt].ls = tree[rt].rs = tree[rt].ms = 1; //修复 
		else
			tree[rt].ls = tree[rt].rs = tree[rt].ms = 0; //破坏 
		return;
	}
	
	int mid = (tree[rt].l + tree[rt].r) >> 1;
	if (t <= mid) insert(rt<<1, t, flag);
	else insert(rt<<1|1, t, flag);
	
	
	tree[rt].ls = tree[rt<<1].ls; //左区间 
	tree[rt].rs = tree[rt<<1|1].rs; //右区间 
	tree[rt].ms = max(max(tree[rt<<1].ms, tree[rt<<1|1].ms), tree[rt<<1].rs+tree[rt<<1|1].ls);
	//父亲区间内的最大区间必定是，左子树最大区间，右子树最大区间，左右子树合并的中间区间，三者中最大的区间值   
	//建议大家在纸上推一遍
	 
	if (tree[rt<<1].ls == tree[rt<<1].r-tree[rt<<1].l+1) //左子树区间满了的话，父亲左区间要加上右孩子的左区间  
		tree[rt].ls += tree[rt<<1|1].ls;
	if (tree[rt<<1|1].rs == tree[rt<<1|1].r-tree[rt<<1|1].l+1) //右子树区间满了的话，父亲右区间要加上左孩子的右区间 
		tree[rt].rs += tree[rt<<1].rs;
}

int query(int rt, int t) {
	if (tree[rt].l == tree[rt].r || (!tree[rt].ms) || tree[rt].ms == tree[rt].r-tree[rt].l+1) //到了叶子节点或者该访问区间为空或者已满都不必要往下走了   
		return tree[rt].ms;
		
	int mid = (tree[rt].l + tree[rt].r) >> 1;
	
	if (t <= mid) {
		if (t >= tree[rt<<1].r - tree[rt<<1].rs+1) //因为t<=mid，看左子树，tree[2*i].r-ree[2*i].rs+1代表左子树右边连续区间的左边界值，如果t在左子树的右区间内，则要看右子树的左区间有多长并返回  
			return query(rt<<1, t) + query(rt<<1|1, mid+1);
		else  //如果不在左子树的右边界区间内，则只需要看左子树
			return query(rt<<1, t);
	}
	
	else {
		if (t <= tree[rt<<1|1].l + tree[rt<<1|1].ls-1) //同理 
			return query(rt<<1|1, t) + query(rt<<1, mid);
		else 
			return query(rt<<1|1, t);
	}
}


int main(void) {
	ios::sync_with_stdio(0); cin.tie(0);
	while (cin >> n >> m) {
		init(1, n, 1);
		while (m--) {
			char ch; int x;
			cin >> ch;
			if (ch == 'D') {
				cin >> x;
				s.push(x);
				insert(1, x, 0);
			} else if (ch == 'Q') {
				cin >> x;
				cout <<query(1, x) << endl;
			} else {
				if (!s.empty()) {
					x = s.top();
					s.pop();
					insert(1, x, 1);
				}
			}
		}
	}
	return 0;
}