题意

给定一棵树,初始时非叶节点均为无色,叶节点会是红色、蓝色或无色。小红和小蓝轮流给无色叶子染色(小红染红色,小蓝染蓝色,小红先染)。所有叶子染完后,非叶节点的颜色将被逐一确定:一个非叶节点的颜色是它所有儿子的颜色中出现较多的那个(保证有奇数个儿子)。最后,根是谁的颜色谁就获胜。求小红是否能赢,若能赢,求出第一步选择哪些叶子能赢。

$n\leq10^6$

题解

一个点有四种情况:

  1. 必红。
  2. 必蓝。
  3. 必先手。
  4. 必蓝,但是有先手优势可以耗蓝操作。

$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;
}
EOF

评论

暂无评论

发表评论

可以用@mike来提到mike这个用户,mike会被高亮显示。如果你真的想打“@”这个字符,请用“@@”。

博客信息

作者
Jayun
时间
2023-11-02 17:13:53