题意:给你n个节点和每个节点的左右孩子的一棵树,问你是不是一颗完全二叉树。
思路:首先要明白怎样是一颗完全二叉树,完全二叉树:叶节点只能出现在最下层和次下层,并且最下面一层的结点都集中在该层最左边的若干位置的二叉树。
解题思路:
先找到树的根节点,然后用类似广搜的方法,确保前一层都放满后再去找下一层。具体做法是用一个队列存放某一层的节点,用cnt表示当前已经遍历过节点的数量,一旦cnt==n(树的节点总数),则表示是完全二叉树,否则在cnt==n之前一旦出现某个节点没孩子的情况,就不是完全二叉树了。
代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
const int maxn = 105;
char str1[maxn], str2[maxn];
int lch[maxn], rch[maxn], lev[maxn];
bool book[maxn];
int n, root;
void solve()
{
queue<int> q;
q.push(root);
bool ok = 1;
int cnt = 1;
int u;
while(!q.empty())
{
u = q.front(); q.pop();
if(lch[u] != -1) q.push(lch[u]), cnt++;
else
{
ok = 0;
break;
}
if(cnt == n) break;
if(rch[u] != -1) q.push(rch[u]), cnt++;
else
{
ok = 0;
break;
}
if(cnt == n)
break;
}
if(ok) printf("YES %d\n", rch[u] == -1 ? lch[u] : rch[u]);
else printf("NO %d\n", root);
}
int main(void)
{
while(cin >> n)
{
memset(book, 0, sizeof(book));
memset(lch, -1, sizeof(lch));
memset(rch, -1, sizeof(rch));
for(int i = 0; i < n; i++)
{
scanf(" %s %s", str1, str2);
int len1 = strlen(str1);
int len2 = strlen(str2);
int num1 = 0, num2 = 0;
if(len1 == 1 && str1[0] == '-') num1 = -1;
else
{
for(int j = 0; j < len1; j++)
num1 = num1*10+str1[j]-'0';
book[num1] = 1;
}
if(len2 == 1 && str2[0] == '-') num2 = -1;
else
{
for(int j = 0; j < len2; j++)
num2 = num2*10+str2[j]-'0';
book[num2] = 1;
}
lch[i] = num1;
rch[i] = num2;
}
for(int i = 0; i < n; i++)
if(!book[i])
{
root = i;
break;
}
if(n == 1) puts("YES 0");
else solve();
}
return 0;
}