SWUST OJ 987: 输出用先序遍历创建的二叉树是否为完全二叉树的判定结果

987: 输出用先序遍历创建的二叉树是否为完全二叉树的判定结果

题目链接-987: 输出用先序遍历创建的二叉树是否为完全二叉树的判定结果

解题思路
完全二叉树的特点：叶子结点只能出现在最下层和次下层，且最下层的叶子结点集中在树的左部。

• 如果树为空，则直接输出Y，否则层序遍历二叉树进行判断
• 如果一个结点左右孩子都不为空，将其左右孩子入队列
• 如果遇到一个结点，左孩子不存在，右孩子存在，则该树一定不是完全二叉树
• 如果它的左孩子存在，右孩子不存在，就做个标记，若这个左孩子有后代则不是完全二叉树
• 若直至队列为空以上两个条件皆不存在，则说明该二叉树为完全二叉树
• 具体操作见代码

附上代码

#pragma GCC optimize("-Ofast","-funroll-all-loops")
#include<bits/stdc++.h>
#define int long long
#define lowbit(x) (x &(-x))
#define endl '\n'
using namespace std;
const int INF=0x3f3f3f3f;
const int dir[4][2]={-1,0,1,0,0,-1,0,1};
const double PI=acos(-1.0);
const double e=exp(1.0);
const double eps=1e-10;
const int M=1e9+7;
const int N=2e5+10;
typedef long long ll;
typedef pair<int,int> PII;
typedef unsigned long long ull;
struct node{
	char data;
	node *l,*r;
}*p;
void build(node *&p){
	char c;
	cin>>c;
	if(c=='#')
		p=NULL;
	else{
		p=new node;
		p->data=c;
		build(p->l);
		build(p->r);
	}
}
queue<node*> q;
bool judge(node *&p){
	q.push(p);
	bool ass=0;
	while(!q.empty()){
		node *tmp=q.front();
		q.pop();
		if((ass&&(tmp->l!=NULL||tmp->r!=NULL))||(tmp->l==NULL&&tmp->r!=NULL))
			return 0;
		if(tmp->l!=NULL)
			q.push(tmp->l);
		if(tmp->r!=NULL)
			q.push(tmp->r);
		if(tmp->l!=NULL&&tmp->r==NULL)
			ass=1;//做标记
	}
	return 1;
}
signed main(){
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);

	build(p);
	if(p==NULL) cout<<"Y";
	else cout<<(judge(p)?"Y":"N");
	return 0;
}