HDU1540 Tunnel Warfare

知识点：线段树，栈

前面做了两个线段树上二分的题了，这个题其实思路就比较自然了，还是用线段树上二分，算法竞赛上面用的好像是类似维护最大子列和的思路，还没看我就觉得不如线段树上二分，

首先这个题是一个单点修改，区间查询的二分，维护是一个01序列的区间和，非常简单的线段树，然后是二分部分，我们首先可以直接记录每个点是0还是1，如果是0那么就直接不用查询了，答案直接是0，这也是我觉得线段树二分需要注意的一个点，就是要提前处理好边界情况，而不是用二分的代码自动判断，接下来那么查询点肯定是1了，我们就是需要寻找它左边第一个0，右边第一个0的位置，然后两个位置相减再减一就是想要的答案，我们先把两边的答案初始化为0和n+1，这就是两边分别没有零的时候的答案，这样左边有0的话，我们才去二分，没有零的话我们就直接拿上面的0来用，找左边的第一个挨着的0，那么就是用查询函数确定左边的0的个数，用区间长度减去区间的1的个数就行了，这样我们就是在整个线段树上找第左边0的个数的0的位置，右边的第一个挨着的0显然是第左边0的个数+1的0的位置，这样用线段树二分就很方便的找到两边的第一个0的位置，就能得出答案

算是一个基础的线段树上二分的练习

#include <bits/stdc++.h>

using namespace std;

const int N = 5e4 + 5;

struct segt {
	int l, r, sum;
} t[N * 4];

int n, m, a[N];

void build(int p, int l, int r) {
	t[p].l = l; t[p].r = r;
	if (l == r) { t[p].sum = 1; a[l] = 1; return; }
	int mid = (l + r) / 2;
	build(p * 2, l, mid);
	build(p * 2 + 1, mid + 1, r);
	t[p].sum = t[p * 2].sum + t[p * 2 + 1].sum;
}

void change(int p, int x, int v) {
	if (t[p].l == t[p].r) { t[p].sum = v; return; }
	int mid = (t[p].l + t[p].r) / 2;
	if (x <= mid) change(p * 2, x, v);
	else change(p * 2 + 1, x, v);
	t[p].sum = t[p * 2].sum + t[p * 2 + 1].sum;
}

int ask(int p, int l, int r) {
	if (l <= t[p].l && t[p].r <= r) return t[p].sum;
	int sum = 0, mid = (t[p].l + t[p].r) / 2;
	if (l <= mid) sum += ask(p * 2, l, r);
	if (r > mid) sum += ask(p * 2 + 1, l, r);
	return sum;
}

int query(int p, int res) {
	if (t[p].l == t[p].r) return t[p].l;
	int p1 = p * 2;
	int len = t[p1].r - t[p1].l + 1;
	if (res <= len - t[p1].sum) return query(p * 2, res);
	else return query(p * 2 + 1, res - len + t[p1].sum);
}

int main() {
	while (scanf("%d%d", &n, &m) != EOF) {
		build(1, 1, n);
		stack<int> st;
		while (m--) {
			char ch;
			cin >> ch;
			int x;
			if (ch == 'D') {
				scanf("%d", &x);
				change(1, x, 0);
				a[x] = 0;
				st.push(x);
			} else if (ch == 'Q') {
				scanf("%d", &x);
				if (!a[x]) { printf("0\n"); continue; }
				int left = (x > 1 ? ask(1, 1, x - 1) : 0);
				int right = (x < n ? ask(1, x + 1, n) : 0);
				int l = 0, r = n + 1;
				if (left < x - 1) l = query(1, x - 1 - left);
				if (right < n - x) r = query(1, x - 1 - left + 1);
				printf("%d\n", r - l - 1);
			} else {
				if (!st.empty()) {
					a[st.top()] = 1;
					change(1, st.top(), 1);
					st.pop();
				}
			}
		}
	}
	return 0;
}