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

 

 

 

 

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值