【Hdu】1540 Tunnel Warfare（线段树|区间合并）

题目大意：有N个村子排成一条直线，每个村子都连接了它的左右两个村子（除了最左边和最右边的外），有3种操作，（1）"D x"，表示将第x个村子摧毁。（2）"Qx"，表示查询与第x个村子直接和间接相连的村子有多少个。（3）"R"，表示将上一次摧毁的村子复原。

这边有一个需要注意的点是村子可以多次被摧毁，但只需要一次便可恢复。

思路：

这边我只用两个数组lsum，和rsum

其中作用分别是表示左边界向右连续的最大长度，右边界向左连续的最大长度

然后先询问的函数query中，所采用的是不断缩放区间，直到最后成为一个点，并且缩放后区间的lsum和rsum的意义也发生改变

lsum表示的是左边界向左连续的最大长度，rum表示右边界向右连续的最大长度，这样最后对于询问的点q，lsum表示的就是不包括该点向左连续的最大长度，rsum表示的就是不包括该点向右连续的最大长度。

有了这个思想基础之后，剩下的便是转化了，搞了好久。详细看代码吧。

#define  _CRT_SECURE_NO_WARNINGS
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<stack>
using namespace std;
#define MAX 50005
#define ls rt<<1
#define rs ls|1
#define m (l+r)>>1

int lsum[MAX << 2];
int rsum[MAX << 2];
int bb[MAX];
int ans;

void uprt(int len, int rt)
{
	lsum[rt] = lsum[ls];
	rsum[rt] = rsum[rs];
	if (lsum[rt] == len - (len >> 1))
		lsum[rt] += lsum[rs];
	if (rsum[rt] == (len >> 1))
		rsum[rt] += rsum[ls];
}

void updata(int q, int c, int l, int r, int rt)
{
	if (l == r)
	{
		lsum[rt] = rsum[rt] = c;
		bb[l] = c;
		return;
	}
	int mid = m;
	if (q <= mid)
		updata(q, c, l, mid, ls);
	else
		updata(q, c, mid + 1, r, rs);
	uprt(r - l + 1, rt);
}

void query(int q, int l, int r, int rt)
{
	if (l == r)
	{
		ans = (lsum[rt] + rsum[rt])*bb[l] + bb[l];
		return;
	}
	int mid = m;
	int len = r - l + 1;
	int a, b;
	if (q <= mid)
	{
		a = rsum[ls];
		b = lsum[ls];
		if (rsum[ls] != 0)
			rsum[ls] = lsum[rs];
		if (lsum[rs] == (len >> 1))
			rsum[ls] += rsum[rt];
		lsum[ls] = lsum[rt];
		query(q, l, mid, ls);
		rsum[ls] = a;
		lsum[ls] = b;
	}
	else
	{
		a = rsum[rs];
		b = lsum[rs];
		if (lsum[rs] != 0)
			lsum[rs] = rsum[ls];
		if (rsum[ls] == len - (len >> 1))
			lsum[rs] += lsum[rt];
		rsum[rs] = rsum[rt];
		query(q, mid + 1, r, rs);
		rsum[rs] = a;
		lsum[rs] = b;
	}
}

void reset(int l, int r, int rt)
{
	if (l == r)
		return;
	int mid = m;
	reset(l, mid, ls);
	reset(mid + 1, r, rs);
	uprt(r - l + 1, rt);
}

void build(int l, int r, int rt)
{
	if (l == r)
	{
		rsum[rt] = lsum[rt] = 1;
		bb[l] = 1;
		return;
	}
	int mid = m;
	build(l, mid, ls);
	build(mid + 1, r, rs);
	uprt(r - l + 1, rt);
}

int main()
{
	int n, k;
	char str[2];
	int a;
	while (~scanf("%d%d", &n, &k))
	{
		stack<int>pre;
		build(1, n, 1);
		for (int i = 0; i < k; i++)
		{
			scanf("%s", str);
			if (str[0] == 'D')
			{
				scanf("%d", &a);
				pre.push(a);
				updata(a, 0, 1, n, 1);
				continue;
			}
			if (str[0] == 'Q')
			{
				scanf("%d", &a);
				lsum[1] = 0;
				rsum[1] = 0;
				query(a, 1, n, 1);
				printf("%d\n", ans);
				continue;
			}
			if (!pre.empty())
			{
				while (pre.empty() && bb[pre.top()])
					pre.pop();
				updata(pre.top(), 1, 1, n, 1);
				pre.pop();
			}
		}
	}
}