PAT (Advanced Level) 1110——完全二叉树 ( Ps.为什么总是栽到“多位下标”上)

题目传送门

在构建树的时候，记录每个结点的父节点
这样很容易就可以找到整棵树的根节点

之后使用bfs判断是否为满二叉树
我分成四种情况考虑：

• 左右儿子双全：safe，不能否认ta是满二叉树
• 有左儿子没有右儿子：safe，不能否认ta是满二叉树，但是此后队列中的结点都不能有孩子
• 没有左儿子有右儿子：绝对不是满二叉树
• 没有儿子：safe，不能否认ta是满二叉树，但是此后队列中的结点都不能有孩子

Tip

唯一需要注意的坑：
下标可能是两位数，不能只用一个字符储存

一开始测试点显示运行错误，我还以为是队列的空间没有开够
最后才发现是下标处理有问题

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
using namespace std;

const int N = 21;
int n;
struct node {
	int lc, rc;
	int fa;
};
node T[N + 1];
int rt;

bool hv(int x) { return x != -1; }

int bfs(int rt) {
	int Q[N * 2];   
	int tou = 0, wei = 0;
	int canHaveChild = 1;
	Q[++wei] = rt;
	while (tou < wei) {
		int x = Q[++tou];
		//无左儿子有右儿子
		if (!hv(T[x].lc) && hv(T[x].rc)) {
			//printf("%d does not have left child but have right child.\n",x);
			return -1;
		}
		if (canHaveChild) {
			if (!(hv(T[x].lc) && hv(T[x].rc))) {
				canHaveChild = 0;
				//之后队列中的结点都不能有儿子
			}
		}
		else {
			if (hv(T[x].lc) || hv(T[x].rc)) {
				//printf("%d can not have a child.\n", x);
				return -1;
			}
		}
		if (T[x].lc != -1) Q[++wei] = T[x].lc;
		if (T[x].rc != -1) Q[++wei] = T[x].rc;
	}
	return Q[wei];
}

int ToInt(char *s) {
	int ans = 0;
	for (int i = 0; i < strlen(s); ++i)
		ans = ans * 10 + s[i] - '0';
	return ans;
}

int main()
{
	scanf("%d\n",&n);
	for (int i = 0; i < n; ++i) T[i].fa = -1;
	
	//注意下标可能是两位数，不能只用一个字符储存
	char l[3], r[3];
	for (int i = 0; i < n; ++i) {
		scanf("%s %s",&l,&r);
		if (l[0] == '-') T[i].lc = -1;
		else {
			T[i].lc = ToInt(l);
			T[T[i].lc].fa = i;
		}
		if (r[0] == '-') T[i].rc = -1;
		else {
			T[i].rc = ToInt(r);
			T[T[i].rc].fa = i;
		}
	}
	for (int i=0;i<n;++i) 
		if (T[i].fa == -1) {
			rt = i;
			break;
		}
	int lst = bfs(rt);
	if (lst!=-1) printf("YES %d",lst);
	else printf("NO %d",rt);
	return 0;
}