题目描述
输入一棵二叉树,求该树的深度。从根结点到叶结点依次经过的结点(含根、叶结点)形成树的一条路径,最长路径的长度为树的深度。
class Solution:
def TreeDepth(self, pRoot):
# write code here
if pRoot==None:
return 0
depth=0
left_h=0
right_h=0
if pRoot.left:
left_h=self.TreeDepth(pRoot.left)+1
if pRoot.right:
right_h=self.TreeDepth(pRoot.right)+1
if not pRoot.left and not pRoot.right:
depth=1
else:
depth=left_h if(left_h>right_h) else right_h
return depth
简易版:
class Solution:
def TreeDepth(self, pRoot):
if pRoot==None:
return 0
return max(self.TreeDepth(pRoot.left),self.TreeDepth(pRoot.right))+1