给定一个二叉树,找出其最大深度。
二叉树的深度为根节点到最远叶子节点的最长路径上的节点数。
说明: 叶子节点是指没有子节点的节点。
示例:
给定二叉树 [3,9,20,null,null,15,7]
,
3 / \ 9 20 / \ 15 7
返回它的最大深度 3
递归的方法没啥说的,从数据结构这门课上就讲了好多遍。 最大深度就是左子树和右子树取最大值后+1
class Solution:
def maxDepth(self, root: TreeNode) -> int:
if root == None:
return 0
left = self.maxDepth(root.left)
right = self.maxDepth(root.right)
return 1 + max(left,right)
BFS: 用BFS就省去了写递归,直接一个while循环加一个for循环搞定,while循环加for循环也是我刚学会的。。。 思路清晰,代码简单。
class Solution:
def maxDepth(self, root: TreeNode) -> int:
if root == None:
return 0
que=[]
que.append(root)
level=1
while que: # while + for
length = len(que)
for i in range(length):
node=que.pop(0)
if node.left != None:
que.append(node.left)
if node.right != None:
que.append(node.right)
if que:
level=level+1
return level
DFS: 判断点:当该节点是叶子节点的时候,比较当前的level和之前保留的最大深度,取最大值赋给最大深度。
class Solution:
def max(self,a,b):
if a>b:
return a
else:
return b
def dfs(self,root,level):
if root.left == None and root.right == None: # 当前节点是叶子节点
self.maxx = self.max(self.maxx,level)
else:
if root.left != None:
self.dfs(root.left,level+1)
if root.right != None:
self.dfs(root.right,level+1)
def maxDepth(self, root: TreeNode) -> int:
if root == None:
return 0
self.maxx=0
if root.left != None:
self.dfs(root.left,2)
if root.right != None:
self.dfs(root.right,2)
if root.left == None and root.right == None:
self.maxx=1
return self.maxx