[LeetCode 110] - 平衡二叉树 (Balanced Binary Tree)

问题

给出一棵二叉树，判断它是否在高度上是平衡的。

对于本问题，高度上平衡的二叉树定义为：每个节点的两棵子树的深度差永远不大于1的一棵二叉树。

 

初始思路

根据定义，思路应该比较直接：递归计算每个节点左右子树的深度，只要发现一次深度差大于1的情况，即可终止递归返回不平衡的结果。最终代码如下：

class Solution
    {
    public:
        bool isBalanced(TreeNode *root)
        {
            if(!root)
            {
                return true;
            }
            
            isBalanced_ = true;
            
            GetDepth(root);
            
            return isBalanced_;
        }
        
    private:
        int GetDepth(TreeNode* node)
        {
//已经不平衡后不需要再继续了，返回的深度也没用了，随便返回一个0即可
            if(!isBalanced_)
            {
                return 0;
            }
            
            if(!node->left && !node->right)
            {
                return 1;
            }
            
//当前节点贡献1的深度
            int left  = 1;
            int right = 1;
            
            if(node->left)
            {
                left += GetDepth(node->left);
            }
            
            if(node->right)
            {
                right += GetDepth(node->right);
            }
            
            if(left > right + 1 || right > left + 1)
            {
                isBalanced_ = false;
                
                return 0;
            }
            
            return left > right ? left : right;
        }
        
        bool isBalanced_;
        
    };