计算一棵树的高度
int height(TreeNode*p)
{
if(p==nullptr)
return 0;
return max(height(p->left),height(p->right))+1;
}
真正确定上下界的检查函数
bool For_use(TreeNode*p,long long int low,long long int top)
{
if(p==nullptr)
return true;
//检查数值
if(p->val<=low)
{
return false;
}
if(p->val>=top)
{
return false;
}
//检查平衡因子
if(abs(height(p->left)-height(p->right))>1)
{
return false;
}
return For_use(p->left,low,p->val)&&For_use(p->right,p->val,top);
}
调用的接口函数
bool Check_the_tree_is_banlanced_or_not(TreeNode*p)
{
if(p==nullptr)
return true;
return For_use(p,INT_MIN,INT_MAX);
}
整体代码
#include<iostream>
#include<vector>
#include<climits>
#include<cstdbool>
#include<algorithm>
using namespace std;
typedef struct TreeNode
{
int val;
TreeNode *left;
TreeNode *right;
TreeNode(int x) : val(x), left(NULL), right(NULL) {}
} TreeNode;
int height(TreeNode*p)
{
if(p==nullptr)
return 0;
return max(height(p->left),height(p->right))+1;
}
bool For_use(TreeNode*p,long long int low,long long int top)
{
if(p==nullptr)
return true;
//检查数值
if(p->val<=low)
{
return false;
}
if(p->val>=top)
{
return false;
}
//检查平衡因子
if(abs(height(p->left)-height(p->right))>1)
{
return false;
}
return For_use(p->left,low,p->val)&&For_use(p->right,p->val,top);
}
bool Check_the_tree_is_banlanced_or_not(TreeNode*p)
{
if(p==nullptr)
return true;
return For_use(p,INT_MIN,INT_MAX);
}
TreeNode*create(TreeNode*op)
{
int Ai;
cin>>Ai;
if(Ai<0)
{
op=nullptr;
}
else
{
op=new TreeNode(Ai);
op->left=create(op->left);
op->right=create(op->right);
}
return op;
}
void DFS(TreeNode*op)
{
if(op==nullptr)
return;
cout<<op->val;
DFS(op->left);
DFS(op->right);
}
int main()
{
TreeNode*o=nullptr;
o=create(o);
cout<<"树的高度:"<<height(o)<<endl;
cout<<"判断树是否为平衡二叉树(0为假,1为真)"<<Check_the_tree_is_banlanced_or_not(o);
cout<<endl<<"前序遍历二叉树:";
DFS(o);
return 0;
}
运行截图: