判断一棵树是不是平衡二叉树

题目描述

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

解法

• 最直观的想法是从根节点开始遍历，对每个节点求其左右子树高度，若左右子树高度差大于1，则返回false；否则以递归的方式处理其左子节点和右子节点

class Solution {
public:
    bool IsBalanced_Solution(TreeNode* pRoot) {
        if(pRoot == nullptr)
            return true;
        int left = getDepth(pRoot -> left);
        int right = getDepth(pRoot -> right);
        if(abs(left - right) > 1)
            return false;
        return IsBalanced_Solution(pRoot -> left) && IsBalanced_Solution(pRoot -> right);
    }
    
    int getDepth(TreeNode* root) {
        if(root == nullptr)
            return 0;
        int left = getDepth(root -> left);
        int right = getDepth(root -> right);
        return max(left, right) + 1;
    }
};

• 上面的想法虽然直观，但是从根节点开始求高度，会多次遍历下层节点，因此可以从叶节点开始判断是否是平衡二叉树

class Solution {
public:
    bool IsBalanced_Solution(TreeNode* pRoot) {
        return helper(pRoot) != -1;
    }
    
    int helper(TreeNode* root) {
        if(root == nullptr)
            return 0;
        int left = helper(root -> left);
        if(left == -1)
        	return -1;
        int right = helper(root -> right);
        if(right == -1)
        	return -1;
        return abs(left - right) > 1 ? -1 : max(left, right) + 1;
    }
};