Leetcode-5148 Binary Tree Coloring Game(二叉树着色游戏)

这题还是有点意思的，因为只能从第一个染色节点开始拓展，所以作为后手的B，一定是将A的第一个染色节点的父亲或者两个儿子染色，堵住A的路，这样才有可能获胜。

所以B占据的区域要么是除A占据的子树以外的部分，要么就是A的第一个染色节点的左子树或者右子树，只有这三种情况.

所以我们计算一下A的第一个染色节点的左子树和右子树的节点个数，就可以得知

1、左子树的节点个数，

2、右子树的节点个数

3、A的第一个染色节点的子树的节点个数

4、除A的第一个染色节点的子树的节点其余节点的个数

如果

1、左子树的节点个数比其余地方的节点个数多

2、右子树的节点个数比其余地方的节点个数多

3、彻底放弃A的第一个染色节点的子树，其他区域的节点个数比A的第一个染色节点的子树的节点个数多

就能获胜，除此以外都不能获胜.

 1 class Solution
 2 {
 3     public:
 4         int f(TreeNode* root)
 5         {
 6             if(!root) return 0;
 7             return f(root->left)+f(root->right)+1;
 8         }
 9         TreeNode* preorder(TreeNode *root,int x)
10         {
11             if(!root) return NULL;
12             if(root->val==x)
13                 return root;
14             TreeNode *t1 = preorder(root->left,x);
15             TreeNode *t2 = preorder(root->right,x);
16             if(t1) return t1;
17             else if(t2) return t2;
18             return NULL;
19         }
20         bool btreeGameWinningMove(TreeNode* root, int n, int x)
21         {
22             TreeNode* t = preorder(root,x);
23             int k1 = f(t->left);
24             int k2 = f(t->right);
25             //k1 > n-k1 || k2 > n-k2 || n-k1-k2-1 > k1+k2+1 分别对应题解中三种情况
26             if(max(2*k1,2*k2)>n || n>2*(k1+k2+1))
27                 return true;
28             return false;
29         }
30 };