【PTA】L3-010 是否完全二叉搜索树 (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<cstdio>
#include<queue>
using namespace std;
typedef struct node{
	int data;
	struct node *left,*right;
}Lnode,*tree;
tree creat(tree t,int x){
	if(!t){
		tree t = new Lnode;
		t->data=x;
		t->left=t->right=NULL;
        return t;
	}
	else{
		if(x>t->data) t->left=creat(t->left,x);
		else t->right=creat(t->right,x);	
        return t;
	}
}
int main(){
	int n,x;
	scanf("%d",&n);
	tree t=NULL;
	for(int i=0;i<n;i++){
		scanf("%d",&x);
		t=creat(t,x);
	}
	queue<tree>q;
	q.push(t);
	int emp=0,flag=0;
	while(!q.empty()){
		tree f=q.front();
		q.pop();
		if(f!=NULL){
			q.push(f->left);
			q.push(f->right);
			if(emp) flag=1;	
			printf("%d",f->data);
		}
		else emp=1;		//当出现空结点，后面的必须都是空结点
		if(q.front()&&!q.empty()) printf(" ");
	}
	if(flag) printf("\nNO");
	else printf("\nYES");
	return 0;
}

另一种方法：

#include<cstdio>
#include<algorithm>
#include<queue>
using namespace std;
const int N=100005;
const int inf=0x3f3f3f3f;
typedef long long ll;
int n,m,ans;
struct node{
	int l,r;
}s[N];
void dfs(int root, int x){
	if(x>root){
		if(s[root].l==0) s[root].l=x;
		else dfs(s[root].l,x);
	}
	if(x<root){
		if(s[root].r==0) s[root].r=x;
		else dfs(s[root].r,x);
	}
}
int main()
{
	int i,j,k,x,y,root;	
	scanf("%d",&n);
	for(i=0;i<n;i++){
		scanf("%d",&x);
		if(i==0) root=x;
		else dfs(root,x);
	}
	int flag=0;
	queue<int>q;
	vector<int>v;
	v.push_back(0);
	q.push(root);
	while(!q.empty()){
		int t=q.front();
		v.push_back(t);
		q.pop();
		if(t!=root) printf(" ");
		printf("%d",t);
		if(s[t].l) q.push(s[t].l);
		if(s[t].r) q.push(s[t].r);	
	}
	if(flag==0)
		for(i=1;i<=v.size();i++){
			if(i*2<=n){
				if(v[i]<v[i*2]) flag=0;
				else {
					flag=1;break;
				}
			} 
			if(i*2+1<=n){
				if(v[i]>v[i*2+1]) flag=0;
				else {
					flag=1;break;
				}
			} 		
		}
	if(flag) printf("\nNO");
	else printf("\nYES");
	return 0;
}