mark,这题没做出来。题目意思是给定一棵树,每个节点都有一个数字,代表星星的数量,最少需要移动多少次能够使得每个节点上面的星星数量都为1。
这题的关键是考虑想要定义的递归函数的意义是什么。我们将问题进行分解,一共需要移动多少次,其实等价于每个节点出去或进来的次数的总和。那么我们就可以定义递归函数为针对该节点其出去或进来的次数。这个次数就相当于左右节点的次数 加上 root.val -1。其中 root.val -1 代表着需要更上层节点输入或输出多少。题解中说明可以理解为过载量。
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution(object):
result = 0
def temp(self,root):
if root == None:
return 0
left_value = self.temp(root.left)
right_value = self.temp(root.right)
self.result += abs(left_value) + abs(right_value)
return left_value + right_value + root.val - 1
def distributeCoins(self, root):
"""
:type root: TreeNode
:rtype: int
"""
self.temp(root)
return self.result