LeetCode110-平衡二叉树

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

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

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

示例 1:

给定二叉树 [3,9,20,null,null,15,7]

    3
   / \
  9  20
    /  \
   15   7
返回 true 。

示例 2:

给定二叉树 [1,2,2,3,3,null,null,4,4]

       1
      / \
     2   2
    / \
   3   3
  / \
 4   4
返回 false 。

一、思路

要证明一棵树是平衡二叉树，需要：

• 证明左子树是一颗平衡二叉树
• 证明右子树是一颗平衡二叉树
• 证明左右子树的深度之差，不超过1

那么需要写个函数来计算树的深度

C++代码：

class Solution {
public:
	bool isBalanced(TreeNode* root) {
		if (!root)
			return true;
		int left_depth = 0, right_depth = 0;
		if (root->left)
			left_depth = treeDepth(root->left, 0);
		if (root->right)
			right_depth = treeDepth(root->right, 0);
		int gap = abs(left_depth - right_depth);
		bool ans = isBalanced(root->left) && isBalanced(root->right) && (gap < 2);
		return ans;
	}

	int treeDepth(TreeNode* root, int pre_depth) {
		int cur_depth = pre_depth + 1;
		if (root->left == NULL && root->right == NULL)
			return cur_depth;
		int left_depth = 0, right_depth = 0;
		if (root->left)
			left_depth = treeDepth(root->left, cur_depth);
		if (root->right)
			right_depth = treeDepth(root->right, cur_depth);
		int depth = max(left_depth, right_depth);
		return depth;
	}

	int max(int a, int b) {
		return (a > b) ? a : b;
	}
};

执行效率：

改进：其实也没什么，思路不变，把两子树之差的判断提前一下，利用与操作的短路特性。
C++代码：

class Solution {
public:
	bool isBalanced(TreeNode* root) {
		if (!root)
			return true;
		int left_depth = 0, right_depth = 0;
		if (root->left)
			left_depth = treeDepth(root->left, 0);
		if (root->right)
			right_depth = treeDepth(root->right, 0);
		int gap = abs(left_depth - right_depth);
		bool ans = (gap < 2) && isBalanced(root->left) && isBalanced(root->right);
		return ans;
	}

	int treeDepth(TreeNode* root, int pre_depth) {
		int cur_depth = pre_depth + 1;
		if (root->left == NULL && root->right == NULL)
			return cur_depth;
		int left_depth = 0, right_depth = 0;
		if (root->left)
			left_depth = treeDepth(root->left, cur_depth);
		if (root->right)
			right_depth = treeDepth(root->right, cur_depth);
		int depth = max(left_depth, right_depth);
		return depth;
	}

	int max(int a, int b) {
		return (a > b) ? a : b;
	}
};

执行效率：