代码随想录day17 二叉树part04 LeetCode110.平衡二叉树 257. 二叉树的所有路径 404.左叶子之和

 

110.平衡二叉树

力扣题目链接(opens new window)

给定一个二叉树，判断它是否是高度平衡的二叉树。

本题中，一棵高度平衡二叉树定义为：一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过1。

思路如下：

终止条件1：读取到空，返回0

开始递归，当左右子树的高度差大于1时，返回-1，代表此子树不符合平衡二叉树性质，若暂时要求，则继续递归，直到末尾。最后对根节点进行判断即可。

class Solution:
    def getHeight(self,root: Optional[TreeNode]) -> int:
        if not root:
            return 0
        left = self.getHeight(root.left)
        if left == -1:
            return -1
        right = self.getHeight(root.right)
        if right == -1:
            return -1
        if abs(left - right) > 1:
            return -1
        else:
            return 1 + max(left,right)

    def isBalanced(self, root: Optional[TreeNode]) -> bool:
        if self.getHeight(root) == -1:
            return False
        else:
            return True

257. 二叉树的所有路径

力扣题目链接(opens new window)

给定一个二叉树，返回所有从根节点到叶子节点的路径。

说明: 叶子节点是指没有子节点的节点。

没想出来。

class Solution:
    def traversal(self, cur, path, result):
        path.append(cur.val)  # 中
        if not cur.left and not cur.right:  # 到达叶子节点
            sPath = '->'.join(map(str, path))
            result.append(sPath)
            return
        if cur.left:  # 左
            self.traversal(cur.left, path, result)
            path.pop()  # 回溯
        if cur.right:  # 右
            self.traversal(cur.right, path, result)
            path.pop()  # 回溯

    def binaryTreePaths(self, root):
        result = []
        path = []
        if not root:
            return result
        self.traversal(root, path, result)
        return result

404.左叶子之和

力扣题目链接(opens new window)

计算给定二叉树的所有左叶子之和。

单节点二叉树的根节点和二叉树左叶子都是满足node and not node.left and not node.right的，所以一开始犯迷糊了。然后重新梳理了一下代码结构，大体思想如下：

1. 定义一个递归函数，函数的参数为当前节点。
2. 在递归函数中，判断当前节点是否为空。如果为空，返回 0。
3. 判断当前节点的左子节点是否存在，且为叶子节点（即没有左右子节点）。如果满足条件，将左叶子节点的值累加。
4. 递归调用函数，分别传入当前节点的左子节点和右子节点。
5. 返回左叶子节点的累加值。

根据以上提纲，构建代码如下：

class Solution:
    def sumOfLeftLeaves(self, root: Optional[TreeNode]) -> int:
        suml = 0
        if not root:
            return 0
        if root.left and not root.left.left and not root.left.right:
            suml += root.left.val
        left = self.sumOfLeftLeaves(root.left)
        right = self.sumOfLeftLeaves(root.right)
        return suml + left class Solution:

之前担心单节点二叉树的根节点和二叉树左叶子都是满足node and not node.left and not node.right的，即算法没有办法分辨单节点和左叶子，实际上用到的判断条件并不是这个，并且单节点是满足不了判断条件root.left and not root.left.left and not root.left.right的，此前的担心是多余的。