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