PAT 1110. Complete Binary Tree (25)(判断完全二叉树)

最新推荐文章于 2024-09-15 22:31:42 发布

cillyb

最新推荐文章于 2024-09-15 22:31:42 发布

阅读量299

点赞数 1

分类专栏：树

本文链接：https://blog.csdn.net/CillyB/article/details/79531160

版权

树专栏收录该内容

16 篇文章 0 订阅

订阅专栏

题意：给你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;
}