LeetCode刷题实战366:寻找二叉树的叶子节点
Given a binary tree, collect a tree’s nodes as if you were doing this: Collect and remove all leaves, repeat until the tree is empty.
依次从左到右,每次收集并删除所有的叶子节点
重复如上过程直到整棵树为空
示例
解题
class Solution:
def findLeaves(self, root: TreeNode) -> List[List[int]]:
res = []
def helper(root):
if not root:
return 0
left = helper(root.left)
right = helper(root.right)
height = max(left, right)
if len(res) == height:
res.append([])
res[height].append(root.val)
return height + 1
helper(root)
return res
评论