题目
输入一棵二叉树的根节点,判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过1,那么它就是一棵平衡二叉树。
思路
第一想法:先序遍历 + 判断深度 (从顶至底)
先序遍历每一个节点,并比较左右子树高度,如果有>1则返回false。具体如下:
思路是构造一个获取当前子树的深度的函数 depth(root) ,通过比较某子树的左右子树的深度差 abs(depth(root->left) - depth(root->right)) <= 1 是否成立,来判断某子树是否是二叉平衡树。若所有子树都平衡,则此树平衡。
- 时间复杂度O(NlogN):计算深度函数depth(root)的时间复杂度是O(N),因为需要遍历其下所有结点;每层会调用两次,数的层数为O(logN)
- 空间复杂度O(N):最差情况下(当树退化为链表时),递归深度可达到 N ,需要使用 O(N) 的栈空间。
/**
* 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:
bool isBalanced(TreeNode* root) {
if( root==NULL ){
return true;
}
//先序遍历:先检测当前根结点的深度差,再检测左右子树的
return abs( depth(root->left)-depth(root->right) ) <= 1 && isBalanced(root->left) && isBalanced(root->right);
}
int depth(TreeNode *root){
if( root==NULL ){
return 0;
}
return max( depth(root->left), depth(root->right) ) + 1;
}
};
最优解法:后序遍历 + 剪枝 (从底至顶)
思路是对二叉树做后序遍历,从底至顶返回子树深度,若判定某子树不是平衡树则 “剪枝” ,直接向上返回。
- 时间复杂度 O(N): N 为树的节点数;最差情况下,需要递归遍历树的所有节点。
- 空间复杂度 O(N): 最差情况下(树退化为链表时),递归深度为N,系统递归需要使用 O(N) 的栈空间。
/**
* 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:
bool isBalanced(TreeNode* root) {
return dfs(root) != -1;
}
//dfs返回值为非负整数(根结点深度),或者-1(表示非平衡)
int dfs(TreeNode *root){
if( root==NULL ){
return 0;
}
//后序遍历
//1. 左子树
int ldepth = dfs( root->left );
if( ldepth==-1 ){
return -1; //剪枝:已检测到左子树不平衡,直接终止递归,剩下的右子树、根结点和更高层不用再判断了
}
//2. 右子树
int rdepth = dfs( root->right );
if( rdepth==-1 ){
return -1; //剪枝:已检测到右子树不平衡,直接终止递归,剩下根结点和更高层不用再判断
}
//3. 根结点
//如果左右子树的深度相差不超过1,那就返回当前根结点的深度,否则就返回-1
if( abs(ldepth - rdepth) <= 1 ){
return max(ldepth, rdepth) + 1;
}else{
return -1;
}
}
};