DFS_366_寻找二叉树的叶子节点

最新推荐文章于 2023-09-20 22:01:49 发布

小孟Tec

最新推荐文章于 2023-09-20 22:01:49 发布

阅读量503

点赞数

分类专栏： LeetCode题解文章标签：二叉树 leetcode 算法数据结构 dfs

本文链接：https://blog.csdn.net/m0_38024592/article/details/106590042

版权

LeetCode题解专栏收录该内容

53 篇文章 0 订阅

订阅专栏

文章目录

题目描述
思路

题目描述

在这里插入图片描述

思路

反向记录层的深度，即叶子结点记为0
用Hash Map 记录每层的数据

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
	def findLeaves(self, root: TreeNode) -> List[List[int]]:
		# like layer traverse  反向记录层的深度
		# key: layer
		# val: [layerVal, ...]
		# Time complexity: O(N)
		# Space complexity: O(1)
		layer_dic = collections.defaultdict(list)
		def solve(node):
			if not node: return 0
			layer = max(solve(node.left), solve(node.right)) + 1
			layer_dic[layer].append(node.val)
			return layer
		solve(root)
		return [val for val in layer_dic.values()]