DFS结题即可,实际上DFS和先序遍历是等价的,先序遍历是先读root,在读左右子树,DFS是先不停的去读左子树,然后不断返回到第一个有右子树的结点,读一下右子树,在返回,依次类推,两者是一样的。
class TreeNode:
def __init__(self, x):
self.val = x
self.left = None
self.right = None
#
# 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
#
#
# @param root TreeNode类
# @return int整型
#
class Solution:
def maxDepth1(self , root: TreeNode) -> int:
# write code here
# 进行一次先序遍历,返回最大深度
self.max_depth = 0
self.tra(root,1)
return self.max_depth
def tra(self,root,level):
if root==None:
return
else:
if level>self.max_depth:
self.max_depth=level
self.tra(root.left , level+1)
self.tra(root.right , level+1)
def maxDepth(self , root: TreeNode) -> int:
self.max_depth = 0
self.DFS(root,1)
return self.max_depth
def DFS(self,root,level):
if root:
if self.max_depth<level:
self.max_depth=level
self.DFS(root.left,level+1)
self.DFS(root.right,level+1)
root = TreeNode(1)
root.left = TreeNode(2)
print(Solution().maxDepth(root))