python--lintcode596 · 最小子树

描述

给一棵二叉树, 找到和为最小的子树, 返回其根节点。
输入输出数据范围都在int内。

 

LintCode会打印根节点为你返回节点的子树。保证只有一棵和最小的子树并且给出的二叉树不是一棵空树。

样例

样例 1:

输入:
{1,-5,2,1,2,-4,-5}
输出:1
说明
这棵树如下所示：
     1
   /   \
 -5     2
 / \   /  \
1   2 -4  -5 
整颗树的和是最小的，所以返回根节点1.

样例 2:

输入:
{1}
输出:1
说明:
这棵树如下所示：
   1
这棵树只有整体这一个子树，所以返回1.

这一题也是用分治法来做，不清楚分治法的童鞋可以看看我上一篇说的：https://blog.csdn.net/wenqiwenqi123/article/details/116306532

每一棵树的和=左子树的和+右子树的和+自己。于是拆解完成，剩下的事只需要保存一下最小值来打擂台。

代码：

class TreeNode:
    def __init__(self, val):
        self.val = val
        self.left, self.right = None, None


class Solution:
    """
    @param root: the root of binary tree
    @return: the root of the minimum subtree
    """
    def findSubtree(self, root):
        # write your code here
        def find(root):
            if root is None:
                return 0
            leftvalue=find(root.left)
            rightvalue=find(root.right)
            tree_value=leftvalue+rightvalue+root.val
            if tree_value<self.value:
                self.value=tree_value
                self.result=root
            return tree_value


        self.value=9999999
        self.result=None
        find(root)
        return self.result


Tree = TreeNode(1)
Tree.left = TreeNode(2)
Tree.left.right = TreeNode(5)

Tree.right = TreeNode(3)
s = Solution()
print(s.findSubtree(Tree))