PATA1110_完全二叉树（难度：⭐️）

原创于 2021-07-14 22:38:45 发布 · 110 阅读

0 ·

CC 4.0 BY-SA版权

PAT甲级刷题专栏收录该内容

124 篇文章

订阅专栏

该代码实现了一个从输入构建二叉树的过程，并通过前序遍历为节点分配从1到N的序号。如果所有节点都有子节点，则输出YES和最后一个节点的序号，否则输出NO和根节点的序号。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

由于没有孩子节点的值是0，所以我命名的序号是1-N，而不是题目给的0-N；所以最后要减去1。

#include <bits/stdc++.h>
using namespace std;
int n, flag = 0;
bool vis[30];
struct node {
    int val, lc, rc, pos;
}tree[30];
unordered_map<int, int> mp;
void posName (int root, int pos) {
    if (root == 0) return;
    tree[root].pos = pos;
    mp[pos] = root;
    if (pos > n) flag = 1;
    posName(tree[root].lc, pos * 2);
    posName(tree[root].rc, pos * 2 + 1);
}
int main() {
    int root;
    string a, b;
    scanf ("%d", &n);
    for (int i = 1; i <= n; i++) {
        cin >> a >> b;
        if (a[0] != '-') {
            tree[i].lc = stoi(a) + 1;
            vis[stoi(a) + 1] = true;
        }
        else tree[i].lc = 0;
        if (b[0] != '-') {
            tree[i].rc = stoi(b) + 1;
            vis[stoi(b) + 1] = true;
        }
        else tree[i].rc = 0;
    }
    for (int i = 1; i < 30; i++) {
        if (vis[i] == false) {
            root = i;
            break;
        }
    }
    posName(root, 1);
    if (flag == 0) printf ("YES %d", mp[n] - 1);
    else printf ("NO %d", root - 1);
}