题意
给定一棵树,初始时非叶节点均为无色,叶节点会是红色、蓝色或无色。小红和小蓝轮流给无色叶子染色(小红染红色,小蓝染蓝色,小红先染)。所有叶子染完后,非叶节点的颜色将被逐一确定:一个非叶节点的颜色是它所有儿子的颜色中出现较多的那个(保证有奇数个儿子)。最后,根是谁的颜色谁就获胜。求小红是否能赢,若能赢,求出第一步选择哪些叶子能赢。
$n\leq10^6$ 。
题解
一个点有四种情况:
- 必红。
- 必蓝。
- 必先手。
- 必蓝,但是有先手优势可以耗蓝操作。
令 $r,b$ 分别表示字节点红蓝情况,那四种情况就是: $r>b,b>r,b=r,b=r+1$ 。
代码
const int N = 1e5 + 5;
inline ll Read() {
	ll x = 0, f = 1;
	char c = getchar();
	while (c != '-' && (c < '0' || c > '9')) c = getchar();
	if (c == '-') f = -f, c = getchar();
	while (c >= '0' && c <= '9') x = (x << 3) + (x << 1) + c - '0', c = getchar();
	return x * f;
}
namespace Main {
	int n;
	vector <int> e[N];
	int col[N];
	bool leaf[N];
	void dfs(int u){
		int r = 0, b = 0; leaf[u] = 1;
		for (int v: e[u]) {
			leaf[u] = 0;
			dfs(v);
			if (col[v] == 0) ++r;
			else if (col[v] != -1) ++b;
		}
		if (r == b) return;
		if (r > b) col[u] = 0;
		else if (b == r + 1) col[u] = 2;
		else col[u] = 1;
	}
	int Ans[N], ans;
	void calc(int u) {
		if (col[u] != -1 && col[u] != 2) return;
		if (leaf[u]) Ans[++ans] = u;
		for (int v: e[u]) calc(v);
	}
	int main () {
		for (int t = Read(); t--; ) {
			n = Read();
			for (int i = 1; i <= n; i++) e[i].clear();
			for (int i = 1; i <= n; i++) {
				int fa = Read();
				if (fa) e[fa].emplace_back(i);
			}
			for (int i = 1; i <= n; i++) col[i] = Read();
			dfs (1);
			int r = 0, b = 0;
			for (int v: e[1]) {
				if (col[v] == 0) ++r;
				else if (col[v] != -1) ++b;
			}
			ans = 0;
			if (r > b) {
				for (int i = 1; i <= n; i++)
					if (leaf[i] && col[i] == -1) Ans[++ans] = i;
			} else if (r == b) {
				calc(1);
				sort (Ans + 1, Ans + 1 + ans);
			} else {
				puts("-1"); return 0;
			}
			printf ("%d ", ans);
			for (int i = 1; i <= ans; i++) printf ("%d ", Ans[i]); putchar(10);
		}
		return 0;
	}
}
int main () {
//	freopen(".in", "r", stdin);
//	freopen(".out", "w", stdout);
	Main::main();
	return 0;
}

 BZOJ5356
BZOJ5356