剑指offer：二叉树的深度（Python）

题目描述

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

解题思路

两种解法，一种是第一时间的思路，即获得根节点到每个叶节点的深度，取其中最长的返回：

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》中所不建议的方式，所以就到此为止了~