描述
给一棵二叉树, 找到和为最小的子树, 返回其根节点。
输入输出数据范围都在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))