7-43 是否完全二叉搜索树 (30 分)
将一系列给定数字顺序插入一个初始为空的二叉搜索树(定义为左子树键值大,右子树键值小),你需要判断最后的树是否一棵完全二叉树,并且给出其层序遍历的结果。
输入格式:
输入第一行给出一个不超过20的正整数N;第二行给出N个互不相同的正整数,其间以空格分隔。
输出格式:
将输入的N个正整数顺序插入一个初始为空的二叉搜索树。在第一行中输出结果树的层序遍历结果,数字间以1个空格分隔,行的首尾不得有多余空格。第二行输出YES,如果该树是完全二叉树;否则输出NO。
输入样例1:
9
38 45 42 24 58 30 67 12 51
输出样例1:
38 45 24 58 42 30 12 67 51
YES
输入样例2:
8
38 24 12 45 58 67 42 51
输出样例2:
38 45 24 58 42 12 67 51
NO
#include<bits/stdc++.h>
using namespace std;
const int N=20000;
typedef struct Node{
int val;
struct Node* l;
struct Node* r;
Node(int x):val(x),l(NULL),r(NULL){}
}Tnode,*Tree;
Tree build(Tree T,int x)
{
if(!T){
T=new Node(x);//走到空了,就建里这个节点
return T;//返回节点
}
if(x<T->val) T->r=build(T->r,x);//小于当前节点走左边
else T->l=build(T->l,x);//大于等于走右边
return T;//返回结果
}
vector<int> v;
int flag=1;
void print(Tree T)
{
queue<Tree> q;
q.push(T);
flag=1;
int t=1;
while(!q.empty()){
Tree temp=q.front();
q.pop();
if(temp->l) q.push(temp->l);
if(temp->r) q.push(temp->r);
v.push_back(temp->val);
if(t==1){
if(!temp->l&&temp->r){ //左子树空,但是右子树不空,肯定不是完全二叉树
flag=0;
}
else if(!temp->r){ //如果右子树空了,证明以后的节点的都是叶子节点
t=-1;//改变t进入下面的判断
}
}
else{
if(temp->r||temp->l){//如果以后的节后不是叶子 ,那就证明不是满二叉树
flag=0;
}
}
}
}
int main()
{
Tree T=NULL;
int i,j,n,m,x;
cin>>n;
for(i=1;i<=n;i++){
cin>>x;
T=build(T,x);
}
print(T);
for(i=0;i<v.size();i++){
printf("%d",v[i]);
if(i!=v.size()-1) printf(" ");
}
if(flag){
printf("\nYES\n");
}
else{
printf("\nNO\n");
}
}