LeetCode-Python-1145. 二叉树着色游戏

有两位极客玩家参与了一场「二叉树着色」的游戏。游戏中，给出二叉树的根节点 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

思路：

首先要搞懂题意，题目说的是让你在对手已经选了一个节点X的前提下，

再选一个节点Y，然后把Y【相邻】的【未染色】的节点一个个染色，

所以出于贪心的思想，应该考虑选跟X相邻的节点，这样可以让X可以染的节点数尽可能的小。

因为X无法越过Y已经染色的节点。

 

先找到x对应的节点x_address， 找到之后分类讨论一下：

1. 如果x就是根节点的值，那么情况比较简单，

•  如果左右子树节点数相同为N ，则不管y选左右子节点哪一个，都必输。因为最多选N个，而对手可以选1 + N个
• 如果左右子树节点数不同，那么必赢，因为只要选了节点数多的那一侧则稳赢

2. 如果x不是根节点的值，那么现在有三种可能的选法，能获胜

• 选x节点的父节点
• 选x节点的右子节点
• 选x节点的左子节点

先看第一种情况：

假设X选了3，那么此时必赢，因为只要拿下根节点，就可以让对手只能选3, 6。

假设X选了3， 那么此时Y选4，必赢，因为对手最多选3， 6， 7。

第三种情况是第二种情况的镜像，因此比较简单不再讨论。

所以，只要统计一下整棵树的节点数，X可选的节点数， Y可选的节点数，

分别计算一下三种情况有没有胜利的可能，即可得到答案。

class Solution(object):
    def btreeGameWinningMove(self, root, n, x):
        """
        :type root: TreeNode
        :type n: int
        :type x: int
        :rtype: bool
        """
        self.x_address = 0
        
        def dfs(node): #找x对应的节点的地址
            if not node:
                return 
            if node.val == x:
                self.x_address = node
                return
            dfs(node.left)
            dfs(node.right)
        dfs(root)
        
        def cnt_node_in_subtree(node): #统计以一个节点为根节点的子树里结点的个数
            if not node:
                return 0
            return 1 + cnt_node_in_subtree(node.left) + cnt_node_in_subtree(node.right)
        
        if root.val == x:     
            return cnt_node_in_subtree(root.left) != cnt_node_in_subtree(root.right)
        else:
            x_cnt = cnt_node_in_subtree(self.x_address)
            total_cnt = cnt_node_in_subtree(root) 
            left_cnt = cnt_node_in_subtree(self.x_address.left)
            right_cnt = cnt_node_in_subtree(self.x_address.right)
            return x_cnt < total_cnt - x_cnt or left_cnt > total_cnt - left_cnt or right_cnt > total_cnt - right_cnt
            # 分别对应 选父节点， 选左孩子，选右孩子