题目描述
有两位极客玩家参与了一场「二叉树着色」的游戏。游戏中,给出二叉树的根节点 root
,树上总共有 n
个节点,且 n
为奇数,其中每个节点上的值从 1
到 n
各不相同。
游戏从「一号」玩家开始(「一号」玩家为红色,「二号」玩家为蓝色),最开始时,
「一号」玩家从 [1, n]
中取一个值 x
(1 <= x <= n
);
「二号」玩家也从 [1, n]
中取一个值 y
(1 <= y <= n
)且 y != x
。
「一号」玩家给值为 x
的节点染上红色,而「二号」玩家给值为 y
的节点染上蓝色。
之后两位玩家轮流进行操作,每一回合,玩家选择一个他之前涂好颜色的节点,将所选节点一个 未着色 的邻节点(即左右子节点、或父节点)进行染色。
如果当前玩家无法找到这样的节点来染色时,他的回合就会被跳过。
若两个玩家都没有可以染色的节点时,游戏结束。着色节点最多的那位玩家获得胜利 ✌️。
现在,假设你是「二号」玩家,根据所给出的输入,假如存在一个 y
值可以确保你赢得这场游戏,则返回 true
;若无法获胜,就请返回 false
。
示例:
输入:root = [1,2,3,4,5,6,7,8,9,10,11], n = 11, x = 3 输出:True 解释:第二个玩家可以选择值为 2 的节点。
提示:
- 二叉树的根节点为
root
,树上由n
个节点,节点上的值从1
到n
各不相同。 n
为奇数。1 <= x <= n <= 100
题解
这个题目还是需要思考一下,这个游戏怎么才能赢。按照游戏规则,我们每一步只能染色相邻的结点,所以我们要想赢,最优的方式就是堵住对方染色的通道。那么在对方选择一个节点之后,有几条通道呢?有三条,父节点,左子节点,右子节点。因此我们最多堵住一条路,那条路余下的所有节点都是我们的。因此只要我们算清这三条通道上各有几个节点,最大的节点数是不是大于所有节点数的一半,就可以了。分析到这里答案就很清楚了。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
int leftNum = 0;
int rightNum = 0;
int calculateNum(TreeNode* root){
if(root==NULL)
return 0;
return calculateNum(root->left)+calculateNum(root->right)+1;
}
void dfs(TreeNode* root, int target){
if(root==NULL)
return;
if(root->val==target)
{
leftNum = calculateNum(root->left);
rightNum = calculateNum(root->right);
return;
}
dfs(root->left,target);
dfs(root->right,target);
}
bool btreeGameWinningMove(TreeNode* root, int n, int x) {
dfs(root,x);
int other = n - leftNum - rightNum - 1;
int half = n/2;
if(other>half||leftNum>half||rightNum>half)
return true;
return false;
}
};