输入一棵二叉树,求该树的深度。从根结点到叶结点依次经过的结点(含根、叶结点)形成树的一条路径,最长路径的长度为树的深度。
找到二叉树中的所有路径,然后获取最长的路径的长度。
找到二叉树中所有路径的方法参照:牛客网 二叉树中和为某一值的路径
# -*- coding:utf-8 -*-
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def TreeDepth(self, pRoot):
if not pRoot:
return 0
ans=self.TreePath(pRoot)
# 获取长度
length=0
for i in ans:
if length<len(i):
length=len(i)
return length
def TreePath(self,pRoot):
if not pRoot:
return []
if pRoot and not pRoot.left and not pRoot.right:
return [[pRoot.val]]
ans=[]
left=self.TreePath(pRoot.left)
right=self.TreePath(pRoot.right)
for i in left+right:
ans.append([pRoot.val]+i)
return ans