PAT甲级 1110 Complete Binary Tree

最新推荐文章于 2024-08-22 10:56:53 发布

CharmsGraker

最新推荐文章于 2024-08-22 10:56:53 发布

阅读量432

点赞数

分类专栏：笔记文章标签：算法 c++ 图论

本文链接：https://blog.csdn.net/qq_35332805/article/details/129367926

版权

笔记专栏收录该内容

18 篇文章 0 订阅

订阅专栏

文章介绍了一种使用C++解决完全二叉树层序遍历问题的方法，通过维护一个队列进行层次遍历，利用完全二叉树的特性判断给定树是否合法。在遍历过程中，遇到空指针时检查剩余节点数，如果剩余节点数大于0则表示树不合法，否则合法。

摘要由CSDN通过智能技术生成

题目链接

PAT甲级 1110 Complete Binary Tree

思路

第一次的写法不是很好。
对于这种完全二叉树的层序遍历，比较烦人的就是空孩子使得处理很麻烦。
思来想去还是把空位置也入队比较好。

这样的话，访问到空指针的时机被推迟了一个level
而完全二叉树的叶子之间的层间距不会超过1
所以对于合法的完全二叉树，我们访问到空孩子的时机不会早于访问树中的任何叶节点

这样的话，当访问到空指针的时候：

对于合法的完全二叉树，一定树中所有元素都遍历完了，故有n == 0
若此时仍有n > 0，否则就说明层间有空隙，故不合法。

所以只要在访问到空指针的时候加入我们的判断即可，并记录还未访问的树节点个数。

代码

#include<bits/stdc++.h>

using namespace std;

vector<int> parent;
vector<pair<int, int>> children;

int last = -1;

int main() {
    int n;
    cin >> n;
    children.resize(n + 1, {-1, -1});
    parent.resize(n + 1, -1);
    for(int i = 0; i < n; ++i) {
        string left, right;
        cin >> left >> right;
        if(left != "-") {
            children[i].first = stoi(left);
            parent[stoi(left)] = i;
        }
        if(right != "-") {
            children[i].second = stoi(right);
            parent[stoi(right)] = i;
        }
    }
    int root = -1;
    for(int i = 0; i < n; ++i) {
        if(parent[i] == -1) {
            root = i;
            break;
        }
    }
    assert(root != -1);
    queue<int> q;
    q.push(root);
    bool flag = true;
    
    while(! q.empty() && flag) {
        int size = q.size();
        int pre_is_null = 0;
        while(size--) {
            auto node = q.front();
            q.pop();
            if(node == -1) {
                if(n > 0) {
                    flag = false;
                }
                break;
            }
            last = node;
            --n;
            auto left = children[node].first;
            auto right = children[node].second;
            q.push(left);
            q.push(right);
        }
    }
    if(! flag) {
        cout << "NO " << root;
    } else
        cout << "YES " << last;
    cout << endl;
}