PAT甲级1110

最新推荐文章于 2022-11-03 09:45:48 发布

原创最新推荐文章于 2022-11-03 09:45:48 发布 · 124 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #二叉树

PAT 专栏收录该内容

68 篇文章

订阅专栏

本文介绍PAT甲级1110题目的解决思路，通过构建树结构并使用队列实现广度优先搜索来判断给定树是否为完全二叉树。详细解释了如何确定根节点及验证各节点是否符合完全二叉树特性。

PAT甲级1110

题目大意：给出总结点数n组成一棵树，结点编号从0-n-1接下来给出n行，每行给出第i-1个结点的左右孩子，若为‘-’表示无孩子。要求判断该树是否是完全二叉树。若是则输出YES并且输出最后一个结点，若不是则输出NO并且输出根结点
首先求出根节点，输入数据的时候开辟bool数组每次出现过的数字设置为false，最后剩余的那个true即为根节点。判断CBT可以开辟一个index数组，保存每个节点所在的下标，从1开始。当循环到一个节点时候判断其是否有孩子以及父亲孩子是否是2倍或者2倍+1的关系，若不满足CBT则flag置为false，全部遍历完返回flag。最后一个元素设置last保存最后一个出队的元素即可。

#include <iostream>
#include <vector>
#include <string>
#include <queue>
#include <algorithm>
#define MAXN 22
using namespace std;

struct node{
	int left, right;
};
node tree[MAXN];
int index[MAXN];
int last;
bool CBT(int root,int n){
	int cnt=1;
	queue<int> q;
	q.push(root);
	while(!q.empty()){
		last = q.front();
		index[last]=cnt++;
		q.pop();
		if(tree[last].left!=-1) q.push(tree[last].left);
		if(tree[last].right!=-1) q.push(tree[last].right);
	}
	bool flag = true;
	for(int i=0;i<n;i++){
		if(tree[i].left!=-1&&index[i]*2!=index[tree[i].left]||tree[i].right!=-1&&index[i]*2+1!=index[tree[i].right])
			flag=false;
	}
	return flag;
}
int main(){
	int n,root;
	string tmpleft, tmpright;
	scanf("%d",&n);
	bool isRoot[n];
	fill(isRoot,isRoot+n,true);
	for(int i=0;i<n;i++){
		cin>>tmpleft>>tmpright;
		if(tmpleft=="-") tree[i].left=-1;
		else tree[i].left=stoi(tmpleft),isRoot[tree[i].left]=false;
		if(tmpright=="-") tree[i].right=-1;
		else tree[i].right=stoi(tmpright),isRoot[tree[i].right]=false;
	}
	for(int i=0;i<n;i++)
		if(isRoot[i]){
			root=i;
			break;
		}
	if(CBT(root,n)) printf("YES %d",last);
	else printf("NO %d",root);
	system("pause");
	return 0;
}