LeetCode 663. 均匀树划分（树形DP）

1. 题目

给定一棵有 n 个结点的二叉树，你的任务是检查是否可以通过去掉树上的一条边将树分成两棵，且这两棵树结点之和相等。

样例 1:
输入:     
    5
   / \
  10 10
    /  \
   2   3
输出: True
解释: 
    5
   / 
  10
      
和: 15

   10
  /  \
 2    3

和: 15
 

样例 2:
输入:     
    1
   / \
  2  10
    /  \
   2   20
输出: False
解释: 无法通过移除一条树边将这棵树划分成结点之和相等的两棵子树。
 
注释 :
树上结点的权值范围 [-100000, 100000]。
1 <= n <= 10000

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/equal-tree-partition
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

• 自底向上求得每个节点的子树和，更新于节点的 val
• 遍历检查+剪枝，共计2次遍历

class Solution {
	bool found = false;
	int allsum;
public:
    bool checkEqualTree(TreeNode* root) {
    	allsum = sum(root);
        if(allsum&1) return false;//奇数不可分
    	check(root);
    	return found;
    }
    int sum(TreeNode* root)
    {
    	if(!root) return 0;
    	int l = sum(root->left);
    	int r = sum(root->right);
    	return root->val += l+r;
    }
    bool check(TreeNode* root)
    {
    	if(!root) return false;
        if(found) return true;
    	if(root->left && allsum == 2*root->left->val)
    		return found = true;
    	if(root->right && allsum == 2*root->right->val)
    		return found = true;
    	return check(root->left) || check(root->right);
    }
};

40 ms 31.9 MB

---

我的CSDN博客地址 https://michael.blog.csdn.net/

长按或扫码关注我的公众号（Michael阿明），一起加油、一起学习进步！