题目:
Given a binary tree, determine if it is height-balanced.
For this problem, a height-balanced binary tree is defined as a binary tree in which the depth of the two subtrees of every node never differ by more than 1.
判断一棵树是否是平衡树 每个结点的左右子树高度相差最多为1
Attention:
1. 一旦发现不平衡,递归切断(前后切断);
2. 理解递归的含义,计算当前节点的子树高度(左右子树的判断一定会放到同一个递归中)。递归虽然是调用相同的函数,但是其实都是副本,即使用不同的函数实现。
AC Code:
/** * Definition for binary tree * struct TreeNode { * int val; * TreeNode *left; * TreeNode *right; * TreeNode(int x) : val(x), left(NULL), right(NULL) {} * }; */ class Solution { public: bool isBalanced(TreeNode *root) { //判断一棵树是否是平衡树 每个结点的左右子树高度相差最多为1 //思路:递归判断每个结点。 //Attention: 一旦发现不平衡,递归切断(前后切断);理解递归的含义,计算当前节点的子树高度(左右子树的判断一定会放到同一个递归中)。递归虽然是调用相同的函数,但是其实都是副本,即时不同的函数实现。 bool result = true; isBalanced_helper(root, result); return result; } private: int isBalanced_helper(TreeNode* root, bool& result){ if(root == NULL) return 0; //切断递归,一旦发现树不平衡,结束递归。 if(result == false) return 0; int leftheight = isBalanced_helper(root->left, result); int rightheight = isBalanced_helper(root->right, result); // prevent overwrite of result 同时也是short-circuit if (result == false) return 0; if(abs(leftheight - rightheight) > 1) result = false; //计算当前节点的子树高度。 return max(leftheight, rightheight) + 1; } };