https://leetcode-cn.com/problems/balanced-binary-tree/
题目
给定一个二叉树,判断它是否是高度平衡的二叉树。
本题中,一棵高度平衡二叉树定义为:
- 一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过1。
示例
给定二叉树 [3,9,20,null,null,15,7]
返回 true 。
给定二叉树 [1,2,2,3,3,null,null,4,4]
返回 false
思路
深搜,以当前节点为根的子树是否为平衡二叉树取决于它的左右子树。
- 首先判断当前节点是否平衡(注意是节点不是当前子树),算出当前节点左右子树的最大深度
- 若当前节点平衡,那么深搜左节点和右节点是否平衡,只有都平衡,以当前节点为根节点的树才整体平衡
- 若当前节点不平衡,那这一整颗树都不平衡,返回false
-
if ((dep(root -> left) - dep(root -> right) < 2) return check(root -> left) && check(root -> right); return false;
代码
不过时间上不太行,写了半天才写出个这个效率
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
int dep(TreeNode* root, int depth) {
if (!root) return depth;
return max(dep(root -> left, depth + 1), dep(root -> right, depth + 1));
}
bool check(TreeNode* root, int depth) {
if (!root) return true;
int t1 = dep(root -> left, depth + 1);
int t2 = dep(root -> right, depth + 1);
if (abs(t1 - t2) < 2)
return check(root -> left, depth + 1) && check(root -> right, depth + 1);
return false;
}
bool isBalanced(TreeNode* root) {
if (!root) return true;
return check(root, 0);
}
};