代码随想录算法训练营第十七天 | leetcode 110. 平衡二叉树,257. 二叉树的所有路径,404.左叶子之和
110. 平衡二叉树
题目:给定一个二叉树,判断它是否是高度平衡的二叉树。
本题中,一棵高度平衡二叉树定义为:一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过1。
示例 1:
给定二叉树 [3,9,20,null,null,15,7]
返回true。
题目链接:110-平衡二叉树
class Solution:
def isBalanced(self, root: Optional[TreeNode]) -> bool:
return self.get_hight(root) != -1
def get_hight(self, root):
if not root:
return 0
left = self.get_hight(root.left)
if left == -1:
return -1
right = self.get_hight(root.right)
if right == -1:
return -1
return max(left, right) + 1 if abs(left - right) < 2 else -1
257. 二叉树的所有路径
题目:给定一个二叉树,返回所有从根节点到叶子节点的路径。
题目链接:257-二叉树的所有路径
class Solution:
def binaryTreePaths(self, root: Optional[TreeNode]) -> List[str]:
path = ''
result = []
if not root:
return result
self.travelsal(root, path, result)
return result
def travelsal(self, cur: TreeNode, path: str, result: List[str]) -> None:
path += str(cur.val)
if not cur.left and not cur.right: #当前结点为叶子结点,则输出结果
result.append(path)
if cur.left:
# + '->' 是隐藏回溯
self.travelsal(cur.left, path + '->', result)
if cur.right:
self.travelsal(cur.right, path + '->', result)
404.左叶子之和
题目:计算给定二叉树的所有左叶子之和。
题目链接:404-左叶子之和
class Solution:
def sumOfLeftLeaves(self, root: Optional[TreeNode]) -> int:
if not root:
return 0
left_left_leaves = self.sumOfLeftLeaves(root.left) #左
right_left_leaves = self.sumOfLeftLeaves(root.right)# 右
cur_leftval_sum = 0
if root.left and not root.left.left and not root.left.right: # 判断是否为左叶子
cur_leftval_sum = root.left.val
return cur_leftval_sum + left_left_leaves + right_left_leaves #中,计算根节点左右两结点左叶子之和