563.二叉树的坡度--python

题:给定一个二叉树,计算整个树的坡度。一个树的节点的坡度定义即为,该节点左子树的结点之和和右子树结点之和的差的绝对值。空结点的的坡度是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
发布了56 篇原创文章 · 获赞 7 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/karen17/article/details/88954210