剑指offer 平衡二叉树

剑指offer题型分类及各题的代码及解题思路

1、题目描述

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

2、初始结构定义如下：

class Solution {
public:
    bool IsBalanced_Solution(TreeNode* pRoot) {}
};

3、思路

每个结点最好只遍历一遍，减小复杂度。

我们可以用后序遍历的方式来遍历二叉树的每个结点，在遍历一个结点之前，我们就已经遍历了它的左右子树，只要在遍历每个节点的时候记录他的深度，我们就可以一边遍历一边判断每个结点是否平衡。

注：所谓平衡，即左右子树的深度差绝对值不大于1。

4、C++实现代码如下

class Solution 
{
public:
    bool IsBalanced_Solution(TreeNode* pRoot) 
    {
        int depth = 0;
        return IsBalanced(pRoot, &depth);
    }
    
    bool IsBalanced(TreeNode* pRoot, int* d)
    {
        if(pRoot == NULL)
        {
            *d = 0;
            return true;
        }
        
        int left,right;
        if(IsBalanced(pRoot->left, &left) && IsBalanced(pRoot->right, &right))
        {
            int diff = left - right;
            if(diff >= -1 && diff <= 1)
            {
                *d = 1 + (left > right ? left : right);
                return true;
            }
        }
        return false;
    }
};