[OpenJudge0054]特务会议召开

[OpenJudge0054]特务会议召开

试题描述

在敌占区的特务时常会碰头。敌占区有n个城市，为保证安全，中央经过侦查，只选择了n-1条较安全的道路作为特务们碰头可以走的道路。每次开会，中央会选择特务正在工作当中的最早开始工作的那一个所在城市作为开会地点。为保证特务安全，开会时，中央会派安全小队在每一条特务会经过的道路上进行保护。每次开会时，中央想知道需要保护的道路共有多少条。

有时中央会认为一个城市很重要，于是在这个城市安插一个特务（即使该城已有特务）；有时一个城市的敌人被干掉了，于是召回这个城市的特务（即使城中没有特务）。一开始中央还没有向敌占区派特务。

输入

第一行两个整数n，m，表示城市数和中央的命令数。
接下来n-1行，每行两个整数x和y，表示x和y间有一条特务可以通过的道路。
接下来m行，每行开头一个字符：
若为 + ，接下来一个数w，表示中央在w安插了一名特务。
若为 - ，接下来一个数w，表示城市w的特务全被召回。
若为 ?，表示特务会议召开，问有多少道路需要被保护。
（+，-与数之间有空格）

输出

对于每次特务会议召开，输出一行一个整数，表示需要被保护的道路数。

输入示例

3 6
1 2
1 3
+ 2
+ 1
+ 3
?
- 2
?

输出示例

2
1

数据规模及约定

对于30%的数据，n<=1000,m<=1000;
对于100%的数据，n<=100000,m<=100000;

题解

借这道题学一下虚树的概念和性质。

不难发现这题我们只需要关心安有特务的点（不妨称之为特殊点），好的，那么我们将这些点提取出来，我们还需要关心他们的 lca，但是两两求 lca 显然是 n² 的；而我们又发现如果将这些点按照它们在树上的 dfs 序大小排个序，找相邻两个点的 lca 就好了。没错，这些 lca 和特殊点就构成了虚树。

性质 1：有不超过 2k 个节点（k 是特殊点个数）。显然 lca 个数小于 k，加起来小于 2k。

性质 2：相邻两个点的距离之和（包括第 1 个和最后一个特殊点）等于虚树大小的一半。这个也很显然，自己画画图就好了。

有了性质 2 这题就做完了。

搞个 set，懒得写平衡树了。

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cctype>
#include <algorithm>
#include <set>
using namespace std;

int read() {
	int x = 0, f = 1; char c = getchar();
	while(!isdigit(c)){ if(c == '-') f = -1; c = getchar(); }
	while(isdigit(c)){ x = x * 10 + c - '0'; c = getchar(); }
	return x * f;
}

#define maxn 100010
#define maxm 200010
#define maxlog 20
int n, q, m, head[maxn], nxt[maxm], to[maxm];

void AddEdge(int a, int b) {
	to[++m] = b; nxt[m] = head[a]; head[a] = m;
	swap(a, b);
	to[++m] = b; nxt[m] = head[a]; head[a] = m;
	return ;
}

int Dfs[maxn], clo, Poi[maxn], Fa[maxlog][maxn], dep[maxn];
void build(int u, int fa) {
	Dfs[u] = ++clo; Poi[clo] = u;
	Fa[0][u] = fa;
	for(int i = 1; i < maxlog; i++) Fa[i][u] = Fa[i-1][Fa[i-1][u]];
	for(int e = head[u]; e; e = nxt[e]) if(to[e] != fa) {
		dep[to[e]] = dep[u] + 1;
		build(to[e], u);
	}
	return ;
}

int calc(int a, int b) {
//	printf("in calc: %d %d
", a, b);
	int ans = dep[a] + dep[b];
	if(dep[a] < dep[b]) swap(a, b);
	for(int i = maxlog - 1; i >= 0; i--)
		if(dep[a] - dep[b] >= (1 << i)) a = Fa[i][a];
	for(int i = maxlog - 1; i >= 0; i--)
		if(Fa[i][a] != Fa[i][b]) a = Fa[i][a], b = Fa[i][b];
	int c = a == b ? a : Fa[0][b];
	ans -= (dep[c] << 1);
	return ans;
}

set <int> S;
std :: set <int> :: iterator it, it2;

int main() {
	n = read(); q = read();
	for(int i = 1; i < n; i++) {
		int a = read(), b = read();
		AddEdge(a, b);
	}
	
	build(1, 0);
//	for(int i = 1; i <= n; i++) printf("(%d)%d ", i, Dfs[i]); putchar('
');
//	for(int i = 1; i <= n; i++) printf("%d ", Poi[i]); putchar('
');
	int ans = 0;
	while(q--) {
		char op[2]; scanf("%s", op);
		if(op[0] == '+') {
			int u = read();
			if(S.count(Dfs[u])) continue;
			if(S.size()) {
				it = S.upper_bound(Dfs[u]);
				int a = *it, b;
				if(it == S.end()) a = *(S.begin());
				if(it == S.begin()) {
					it2 = S.end();
					b = *(--it2);
				}
				else b = *(--it);
				a = Poi[a]; b = Poi[b];
				ans -= calc(a, b);
				ans += calc(a, u); ans += calc(u, b);
			}
			S.insert(Dfs[u]);
		}
		if(op[0] == '-') {
			int u = read();
			if(!S.count(Dfs[u])) continue;
			if(S.size()) {
				it = S.upper_bound(Dfs[u]);
				int a = *it, b;
				if(it == S.end()) a = *(S.begin());
				it = S.lower_bound(Dfs[u]);
				if(it == S.begin()) {
					it2 = S.end();
					b = *(--it2);
				}
				else {
					it2 = it;
					b = *(--it2);
				}
				a = Poi[a]; b = Poi[b];
				ans += calc(a, b);
				ans -= calc(a, u); ans -= calc(u, b);
			}
			S.erase(Dfs[u]);
		}
		if(op[0] == '?') printf("%d
", ans >> 1);
	}
	
	return 0;
}
/*
13 1000
1 2
1 3
1 4
1 5
2 6
2 7
6 8
6 9
6 10
4 11
5 12
5 13
?
+ 13
?
+ 11
?
+ 9
?
*/