PTA L3-010 是否完全二叉搜索树

最新推荐文章于 2022-03-23 22:19:08 发布

(xsj)

最新推荐文章于 2022-03-23 22:19:08 发布

阅读量98

点赞数

分类专栏： # 二叉树与二分搜索树文章标签：二叉搜索树

本文链接：https://blog.csdn.net/qq_44791484/article/details/115132280

版权

二叉树与二分搜索树专栏收录该内容

14 篇文章 0 订阅

订阅专栏

该博客介绍了如何使用数组构建二叉搜索树，并通过层序遍历来判断所构建的树是否为完全二叉树。在代码实现中，通过输入序列动态插入节点并更新最大节点索引，然后根据数组下标输出层序遍历结果，最后检查前n个节点是否都在数组前n个位置以确定是否为完全二叉树。

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

题面

在这里插入图片描述

题解

给定序列，我们建立一颗二叉搜索树，这里用数组存树(根节点为n,那么左儿子为2n，右儿子为2n+1)，建树过程，我们每次从根节点开始遍历，遇到大于根节点的就向左递归，否则向右递归，同时我们更新maxn(用来记录最后一个节点所在数组的下标)，这样一颗树就建好了

对于层序遍历，我们直接按数组下标输出即可，判断是否为完全二叉树，我们只需要看前n个点是否在数组前n个位置即可

代码

#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<algorithm>

using namespace std;
const int N = 1e5 + 10;

int n, x;
int tree[N];

int main() {

    cin >> n;
    int maxn = 0;
    for (int i = 0; i < n; i++) {
        cin >> x;
        int now = 1;   //每次从根节点插入
        while (tree[now]) {
            if (x > tree[now]) {
                now = now * 2;
            } else {
                now = now * 2 + 1;
            }
        }
        tree[now] = x;
        if (now > maxn) maxn = now;
    }

    bool flag = true;
    cout << tree[1];
    for (int i = 2; i <= maxn; i++) {
        if (tree[i]) cout << " " << tree[i];
        else flag = false;  
    }
    cout << endl;
    if (flag) cout << "YES" << endl;
    else cout << "NO" << endl;

    return 0;
}