104. 二叉树的最大深度
给定一个二叉树,找出其最大深度。
二叉树的深度为根节点到最远叶子节点的最长路径上的节点数。
说明: 叶子节点是指没有子节点的节点。
示例:
给定二叉树 [3,9,20,null,null,15,7]
,
3
/
9 20
/
15 7
返回它的最大深度 3 。
#递归
class Solution:
def maxDepth(self, root: TreeNode) -> int:
if not root:
return 0
def dfs(node):
if not node:
return 0
left = dfs(node.left)
right = dfs(node.right)
return max(left,right)+1
return dfs(root)
#栈
def maxDepth(self, root: TreeNode) -> int:
if not root:
return 0
depth = 0
stack = []
stack.append((root,1))
while stack:
node,cur_depth = stack.pop()
if node:
depth = max(depth,cur_depth)
stack.append(node.left,cur_depth+1)
stack.append(node.right,cur_depth+1)
return depth
111. 二叉树的最小深度
给定一个二叉树,找出其最小深度。
最小深度是从根节点到最近叶子节点的最短路径上的节点数量。
说明: 叶子节点是指没有子节点的节点。
示例:
给定二叉树 [3,9,20,null,null,15,7]
,
3
/
9 20
/
15 7
返回它的最小深度 2.
#左右孩子均空,根所在深度
#左右孩子均非空,最小深度
#左右孩子有一个为空,不为空的深度
class Solution:
def minDepth(self, root: TreeNode) -> int:
if not root:
return 0
left = self.minDepth(root.left)
right = self.minDepth(root.right)
children = [root.left,root.right]
if all(children):
return min(left,right)+1
else:
return max(left,right)+1
最新评论