题:给定一个二叉树,计算整个树的坡度。一个树的节点的坡度定义即为,该节点左子树的结点之和和右子树结点之和的差的绝对值。空结点的的坡度是0。
法1:前序遍历,每到一个节点,先用一格helper函数计算每个点左右子树的值,再算每个节点的坡度,这样就得到该节点上的坡度。再递归计算它的左右子节点的坡度。这种方式时间复杂度太大了,因为每次都是从当前根节点开始,越靠近叶子的节点被计算的次数越多。
def findTilt(self, root):
"""
:type root: TreeNode
:rtype: int
"""
if root==None:return 0
left_val=self.helper(root.left)
right_val=self.helper(root.right)
tmp=abs(left_val-right_val)
return tmp+self.findTilt(root.right)+self.findTilt(root.left)
def helper(self,root):
if root==None:return 0
return self.helper(root.left)+self.helper(root.right)+root.val
法2:用后序遍历,这种思想在leetcode 110 平衡二叉树中也有用到,从叶子节点开始计算坡度,减少计算
def findTilt(self, root):
"""
:type root: TreeNode
:rtype: int
"""
res=[]
self.helper(root,res)
print(res)
return sum(res)
def helper(self,root,res):
if root==None:return 0
left_val=self.helper(root.left,res)
right_val=self.helper(root.right,res)
res.append(abs(left_val-right_val))
return left_val+right_val+root.val