给一颗二叉树,a选x节点染为红色,问b是否可以选一个节点y染成蓝色,使得按照一定的规则下b win。
规则:每个选手每次可以选择一个已染色的邻居节点染成自己的颜色,如果没有这样的节点,就跳过这一回合。
最后谁的节点多,谁赢。 节点数为奇数。
sol:
给定了x节点,那么为了最大化b能染得节点数,我们只需要考虑三个节点,x的父节点和x的两个子节点。
int p, l, r;
int n, x;
public boolean btreeGameWinningMove(TreeNode root, int n, int x) {
p = l = r = 0;
this.n = n;
this.x = x;
dfs(root);
return Math.max(Math.max(l, r), n - l - r - 1) > n / 2;
}
private int dfs(TreeNode root) {
if (root == null) return 0;
int l = dfs(root.left);
int r = dfs(root.right);
if (root.val == x) {
this.l = l;
this.r = r;
}
return l + r + 1;
}