PAT(Advanced)1110 Complete Binary Tree C++实现

最新推荐文章于 2021-01-26 09:58:01 发布

原创最新推荐文章于 2021-01-26 09:58:01 发布 · 363 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#数据结构 #算法 #二叉树

数据结构与算法专栏收录该内容

87 篇文章

订阅专栏

PAT(Advanced)甲级1110 Complete Binary Tree C++实现

题目链接

1110 Complete Binary Tree

题目大意

从0到N-1个节点，给定每个节点的左右孩子，求给定二叉树是否为完全二叉树，若是则输出YES以及最右边的叶子节点的序号，否则输出NO以及根节点序号

算法思路

根据题意要求和输入数据，先找到二叉树根节点，即没有父亲节点的节点，根据完全二叉树性质和层序遍历定义，对完全二叉树进行层序遍历，将包括空指针的所有的节点入队，若遇到第一个空指针，则后面必然没有结点，即后面若有节点则必然为空，否则不是完全二叉树。

AC代码

/*
author : eclipse
email  : eclipsecs@qq.com
time   : Fri Jun 26 19:02:33 2020
*/
#include <bits/stdc++.h>
using namespace std;

struct Node {
    int left;
    int right;
};

vector<Node> binaryTree;
int root;
int indexValue;

bool levelOrderTraverse() {
    queue<int> q;
    q.push(root);
    while (!q.empty()) {
        int front = q.front();
        q.pop();
        if (front == -1) {
            while (!q.empty()) {
                if (q.front() != -1) {
                    indexValue = root;
                    return false;
                }
                q.pop();
            }
        } else {
            indexValue = front;
            q.push(binaryTree[front].left);
            q.push(binaryTree[front].right);
        }
    }
    return true;
}

int main(int argc, char const *argv[]) {
    int N;
    scanf("%d", &N);
    binaryTree.resize(N);
    vector<bool> children;
    children.resize(N);
    for (int i = 0; i < N; i++) {
        string left, right;
        cin >> left >> right;
        if (left == "-") {
            binaryTree[i].left = -1;
        } else {
            int child = atoi(left.c_str());
            children[child] = true;
            binaryTree[i].left = child;
        }
        if (right == "-") {
            binaryTree[i].right = -1;
        } else {
            int child = atoi(right.c_str());
            children[child] = true;
            binaryTree[i].right = child;
        }
    }
    for (int i = 0; i < children.size(); i++) {
        if (!children[i]) {
            root = i;
            break;
        }
    }
    printf("%s ", (levelOrderTraverse() ? "YES" : "NO"));
    printf("%d", indexValue);
    return 0;
}