1.平衡二叉树简介
平衡二叉树是指空树或者任意结点左右两棵子树的高度差的绝对值不超过1。
例如:
图1:
图2:
图1是一颗平衡二叉树,图2则不是。
2.平衡二叉树判断
2.1递归法
class Solution {
public:
int gethigh(TreeNode* root){//获取任意结点的高度
if(root==nullptr){
return 0;
}
int l_tree=gethigh(root->left);
int r_tree=gethigh(root->right);
int high=max(l_tree,r_tree)+1;
return high;
}
bool isBalanced(TreeNode* root) {
if(root==nullptr){
return true;
}
int l_high=gethigh(root->left);//获取左子树的高度
int r_high=gethigh(root->right);//获取右子树的高度
return abs(l_high-r_high)<=1 && isBalanced(root->left) && isBalanced(root->right);//左右子树高度差的绝对值小于等于1,并且都是平衡二叉树
}
};
2.2非递归法
通过层序遍历判断每个结点是否是平衡二叉树,其中结点的高度也是通过层序遍历来计算。
class Solution {
public:
int gethigh(TreeNode* cur) {
int high = 0;//结点的高度就是以自己为根结点的层数
if (cur == nullptr) {
return high;
}
queue<TreeNode*> que;//辅助队列
que.push(cur);
while (!que.empty()) {
int size = que.size();
for (int i = 0; i < size; i++) {
TreeNode* node;
node = que.front();
que.pop();
if(node->left){
que.push(node->left);
}
if(node->right){
que.push(node->right);
}
}
high++;
}
return high;
}
bool isBalanced(TreeNode* root) {
if (root == nullptr) {
return true;
}
queue<TreeNode*> que;//辅助队列
que.push(root);
while (!que.empty()) {
TreeNode* node;
node = que.front();
que.pop();
int ret1 = gethigh(node->left);//左子树的高度
int ret2 = gethigh(node->right);//右子树的高度
int dealt = abs(ret1 - ret2);//高度差的绝对值
if (node->left) {
que.push(node->left);
}
if (node->right) {
que.push(node->right);
}
if (dealt <= 1) {//有结点满足平衡二叉树,我们就继续判断其它结点
continue;
} else {
return false;
}
}
return true;
}
};
3.总结
本文采用了两种方法实现了判断是否为平衡二叉树。从效率上来看递归的方法会更高效以及简洁,非递归的方法会更方便我们理解解题的过程。
递归:
非递归: