试解leetcode算法题--平衡二叉树

<题目表述>

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

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

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

示例 1:

给定二叉树 :

  3
 / \
9  20
   / \
  15  7

返回 true 。

示例 2:

给定二叉树

   1
  / \
 2   2
/ \
3  3
  / \
4   4

返回 false 。

<原题链接>

https://leetcode-cn.com/problems/balanced-binary-tree

<思路>

我们根据平衡二叉树的定义，对该树做一次后序遍历，并每次判断其左右子树高度之差是否在[-1, 1]之间，为了方便代码书写，可以将其高度之差先平方，再和1做比较。这里利用了两次递归嵌套。

<样例代码>

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
#include<iostream>
#include<math.h>
using namespace std;
struct TreeNode {
	int val;
	TreeNode *left;
	TreeNode *right;
	TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};

class Solution {
private:
	int getHeight(TreeNode* r);	// 求r的高度
public:
	bool isBalanced(TreeNode* root) {
		if (root != NULL)
		{
			bool a = isBalanced(root->left);
			bool b = isBalanced(root->right);
			if (pow(getHeight(root->left) - getHeight(root->right), 2) > 1)
				return false;
			else
				return a && b;
		}
		return true;
	}
};

int Solution::getHeight(TreeNode * r)
{
	if (r == NULL)
		return 0;
	int h1 = getHeight(r->left);
	int h2 = getHeight(r->right);
	return (h1 > h2 ? h1 : h2) + 1;
}