題目描述

輸入一棵二叉樹，判斷該二叉樹是否是平衡二叉樹。

解題思路

首先我們應該知道，平衡二叉樹就是每一個根的左右子樹的高度差都不大於1，接下來看我的解體思路：

1. 實現一個求二叉樹高度的函式；
2. 利用這個函式求出這個二叉樹左右子樹的高度及高度差diffdepth；
3. 判斷這個高度差絕對值是否小於1；
4. 然後遞迴地判斷子樹是否為平衡二叉樹。

不足之處：多次對左右字數求高度，時間複雜度可能會比較高。

實現程式碼

class Solution {
public:
    //定義一個函式求二叉樹的高度
    int depth(TreeNode* Root)
    {
        if(Root==NULL)
            return 0;
        int leftdepth=depth(Root->left);
        int rightdepth=depth(Root->right);
        
        return leftdepth>rightdepth?leftdepth+1:rightdepth+1;
    }
    bool IsBalanced_Solution(TreeNode* pRoot) {
        if(pRoot==NULL)
            return true;
        int leftdepth=depth(pRoot->left);
        int rightdepth=depth(pRoot->right);
        int diffdepth=leftdepth-rightdepth;
        if(diffdepth<-1 || diffdepth>1)
            return false;
        else
            return (IsBalanced_Solution(pRoot->left) && IsBalanced_Solution(pRoot->right));
    }
};