剑指offer 面试题55 python版+解析：二叉树的深度，平衡二叉树

题目描述

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

思路：

1. 树只有一个节点，则深度为1

2. 根节点只有左子树没有右子树，树的深度为左子树深度加1

3.根节点只有右子树没有左子树，树的深度为右子树深度加1

4.根节点既有右子树又有左子树，树的深度为左、右子树深度的较大值加1

用递归来实现

# -*- 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 == None:
            return 0
        left = self.TreeDepth(pRoot.left)
        right = self.TreeDepth(pRoot.right)
        return left+1 if left>right else right+1

题目描述

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

思路：用后序遍历的方式遍历整棵二叉树。遍历某节点的左、右子节点之后，我们可以根据它的左右子节点的深度判断它是不是平衡的，并得到当前节点的深度。

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    def IsBalanced_Solution(self, pRoot):
        # write code here
        if pRoot is None:
            return True
        depth = 0
        return self.IsBalance(pRoot)>=0
    def IsBalance(self, pRoot):
        if pRoot is None:
            return 0
        left = self.IsBalance(pRoot.left)
        right = self.IsBalance(pRoot.right)
        if left<0 or right<0 or abs(left-right)>1:
            return -1
        return max(left, right)+1