将一系列给定数字顺序插入一个初始为空的二叉搜索树(定义为左子树键值大,右子树键值小),你需要判断最后的树是否一棵完全二叉树,并且给出其层序遍历的结果。
输入格式:
输入第一行给出一个不超过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
判断完全二叉树:在建树的时候给每一个节点给一个id,如果为左树,id是父节点id的两倍,右树则是父节点的id两倍+1,如果为完全二叉树,节点id必然是连续整数。
#include<iostream>
#include<cstdio>
#include<queue>
#include<vector>
using namespace std;
struct node{ int left;
int right;
int id;
};
vector<int> vec,vec2,id;
int j=1,N,flag=1;
struct node a[100000];
void build(int root);
void print(int root);
void cmp(int m,int rt);
int main(void)
{
cin>>N;
for(int i=0;i<N;i++)
{ int k;
cin>>k;
vec.push_back(k);
}
int root=vec[0];
a[root].id=1;
build(root);
print(root);
if(flag)
cout<<"YES";
else
cout<<"NO";
return 0;
}
//建树
void build(int root)
{ //每循环一次必然插入一个节点
while(j!=N)
{ cmp(vec[j],root);
j++;
}
return;
}
//插入的数与根节点比较大小
void cmp(int m,int rt)
{ //大于根节点且根的左子节点为空则插入
if(m>rt&&a[rt].left==0)
{a[m].id=2*a[rt].id; //左子节点id为2*父节点id
a[rt].left=m;
return;
}
else if(m>rt&&a[rt].left!=0) //左子节点非空,递归直到找到空的位置
cmp(m,a[rt].left);
else if(m<rt&&a[rt].right==0) //小于根节点且根的右子节点为空则插入
{a[m].id=2*a[rt].id+1; //右子节点id为2*父节点id+1
a[rt].right=m;
return;
}
else if(m<rt&&a[rt].right!=0)//递归
cmp(m,a[rt].right);
}
//层序遍历
void print(int root)
{
queue<int> q;
q.push(root);
while(!q.empty())
{ int t=q.front();
q.pop();
vec2.push_back(t);
id.push_back(a[t].id); //将节点的id重新顺序存入一个向量中
if(a[t].left)
q.push(a[t].left);
if(a[t].right)
q.push(a[t].right);
}
for(int i=0;i<vec2.size();i++)
{ if(id[i]!=i+1) //如果有一个节点的id不等于完全时的id
flag=0;
printf("%d%c",vec2[i],i==vec2.size()-1?'\n':' ');
}
}