题目描述
输入一棵二叉树,求该树的深度。从根结点到叶结点依次经过的结点(含根、叶结点)形成树的一条路径,最长路径的长度为树的深度。
解题思路
两种解法,一种是第一时间的思路,即获得根节点到每个叶节点的深度,取其中最长的返回:
class Solution:
def __init__(self):
self.maxDeep = 0
self.curDeep = 0
def TreeDepth(self, pRoot):
if not pRoot:
return 0
self.curDeep += 1
self.TreeDepth(pRoot.left)
self.maxDeep = self.curDeep if self.curDeep > self.maxDeep else self.maxDeep
self.TreeDepth(pRoot.right)
self.maxDeep = self.curDeep if self.curDeep > self.maxDeep else self.maxDeep
self.curDeep -= 1
这种方案可以解决问题,但是代码不好看,还定义了两个公共变量。第二种解法是自底而上,每当从子节点回退到其父节点时,返回左右子节点中最深的分支值,没有公共变量,代码短小精悍:
def TreeDeep(self, pRoot):
if not pRoot:
return 0
left = self.TreeDeep(pRoot.left)+1
right = self.TreeDeep(pRoot.right)+1
return left if left>right else right
如果想进一步缩短代码高度,那就把left和right的表达式带入第6行代码中。虽然代码高度进一步缩短,但是增加了代码宽度,且降低了代码的可读性,《Clean Code》中所不建议的方式,所以就到此为止了~