题解 | 判断是不是平衡二叉树

/**
 * struct TreeNode {
 *	int val;
 *	struct TreeNode *left;
 *	struct TreeNode *right;
 *	TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 * };
 */
#include <utility>
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param pRoot TreeNode类 
     * @return bool布尔型
     */
    pair<bool,int> balance(TreeNode* pRoot)
    {
        if(!pRoot)
        {
            return make_pair(true,0);
        }
        pair<bool,int> pair1=balance(pRoot->left);
        pair<bool,int> pair2=balance(pRoot->right);
        if(abs(pair1.second-pair2.second)<=1&&pair1.first==true&&pair2.first==true)
        {
            return make_pair(true,pair1.second>pair2.second?pair1.second+1:pair2.second+1);
        }
        return make_pair(false, -1);
    }
    
    bool IsBalanced_Solution(TreeNode* pRoot) {
        if (!pRoot)
        {
            return true;
        }
        return balance(pRoot).first;
    }
};