leetcode110 平衡二叉树

题目大意

给定一个二叉树，判断它是否是高度平衡的二叉树。

本题中，一棵高度平衡二叉树定义为：

一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过1。

解题思路

根据题意进行分析可知，若当前节点构成的树想要为平衡二叉树，则应该满足：（1）左子树和右子树是平衡二叉树；（2）左子树和右子树的高度差不超过1；
因此我们可以设计一个函数，该函数返回两个值：当前节点构成的树是否是平衡二叉树以及当前树的高度；

struct ReturnType
{
	int level;
	bool isB;
	ReturnType(int l, bool b) :level(l), isB(b) {};
};

class Solution {
public:
    bool isBalanced(TreeNode* root) {
        if (!root)
        	return true;
        return judgeBalance(root).isB;
    }

    ReturnType judgeBalance(TreeNode * root){
    	if (root == nullptr)
    		return ReturnType(0, true);
		//得到左子树的返回信息
    	ReturnType left = judgeBalance(root->left);
    	//如果左子树不是平衡二叉树，则当前树必定不是平衡二叉树
    	if (left.isB == false)
    		//因此直接返回，同时这里的level已经失去了意义
    		//因为左子树不是平衡树了，包含该子树的所有树都不是平衡的了，没有必要再去考虑树的高度
    		return ReturnType(left.level, false);
		//右子树同理
    	ReturnType right = judgeBalance(root->right);
    	if (right.isB == false)
    		return ReturnType(right.level, false);
		//如果左右子树的高度相差大于1
    	if (abs(left.level - right.level) > 1)
    		return ReturnType(left.level, false);

    	return ReturnType(max(left.level, right.level) + 1, true);
    }
};