[剑指Offer]二叉树的深度

题目描述

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

解题思路

二叉树Python实现中已经写过,这里直接照搬一下。基本思路是递归,从根节点出发,求其左右子树的深度,取左右子树的深度+1的较大者为当前树的深度(求所有子树深度的时候使用递归)。如果到达叶子结点,那么返回0。

代码

Python(2.7.3)

1
2
3
4
5
6
7
8
9
10
11
12
13
14
# -*- coding:utf-8 -*-
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def TreeDepth(self, pRoot):
# write code here
if pRoot is None:
return 0
llength = self.TreeDepth(pRoot.left)
rlength = self.TreeDepth(pRoot.right)
return max(llength + 1, rlength + 1)

运行时间:32ms
占用内存:5752k