剑指offer的Python实现之二叉树(3)：树的深度

树的深度及其相关题目

目录

树的深度及其相关题目

1. 二叉树的深度

2. 平衡二叉树

---

1. 二叉树的深度

题目描述：

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

解题思路：

        一棵树的深度为其左、右子树的较大值加1。

class Solution:
    def TreeDepth(self, pRoot):
        # write code here
        if not pRoot:
            return 0
        return max(self.TreeDepth(pRoot.left),self.TreeDepth(pRoot.right))+1

2. 平衡二叉树

题目描述：

        输入一颗二叉树，判断该二叉树是否是平衡二叉树。

解题思路：

        平衡二叉树的定义：如果某棵二叉树中任意节点的左、右子树的深度相差都不超过1，那么它就是一棵平衡二叉树。

        由平衡二叉树的定义可知。要判断一棵树是否为平衡二叉树，我们只需在遍历树的每一节点时调用上一题“二叉树的深度”中的TreeDepth函数得到左、右子树的深度，在判断是否每个节点的左、右子树深度都不超过1即可。

class Solution:
    def IsBalanced_Solution(self, pRoot):
        # write code here
        if not pRoot:
            return True
        if abs(self.TreeDepth(pRoot.left)-self.TreeDepth(pRoot.right))>1:
            return False
        return self.IsBalanced_Solution(pRoot.left) and self.IsBalanced_Solution(pRoot.right)
    
    def TreeDepth(self, pRoot):
        if not pRoot:
            return 0
        return max(self.TreeDepth(pRoot.left),self.TreeDepth(pRoot.right))+1