二叉树深度

题目描述

输入一棵二叉树，求该树的深度。从根结点到叶结点依次经过的结点（含根、叶结点）形成树的一条路径，最长路径的长度为树的深度。 

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