PAT1 1064 Complete Binary Search Tree

最新推荐文章于 2024-11-28 19:59:05 发布

LSC_333

最新推荐文章于 2024-11-28 19:59:05 发布

阅读量148

点赞数 1

分类专栏： PAT

本文链接：https://blog.csdn.net/LSC_333/article/details/93602169

版权

PAT 专栏收录该内容

77 篇文章

订阅专栏

本文介绍了一个算法问题：给定一个无重复数字序列，构建唯一完全二叉搜索树并输出其对应的中序遍历。文章通过排序和递归求根节点的方式解决该问题，展示了具体的Python实现代码。

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

题目链接
 我的github

题目大意

给一个无重复的数字序列，则这个序列可以构成一颗唯一的完全二叉搜索树，要求输出这个完全二叉搜索树的中序遍历

输入

每组包含一个测试用例

第一行是一个正整数 $N\leq1000$ 表示有 $N$ 个不重复的数字
第二行是 $N$ 个不重复的数字

输出

对每个测试用例，在一行中输出这个完全二叉搜索树的中序遍历

样例输入

10
1 2 3 4 5 6 7 8 9 0

样例输出

6 3 8 1 5 7 9 0 2 4

解析

由于是一颗完全的二叉搜索树，所以我们先将输入的序列排序，则排完序之后的序列就是这个完全二叉搜索树的中序遍历，接下来就是不断递归求根节点。因为在层次遍历中，如果根节点的索引是index，那么左子树的根节点的索引就是index*2+1，右子树的根节点的索引就是index*2+2 ，所以求根就是关键。

由于是完全二叉搜索树，那么左子树的所有节点都小于根节点，且左子树的节点数要不小于右子树的节点数。
求根的问题就变成了求左子树的节点数的问题，这样问题就简化了很多。如何求左子树的节点数，其实画下完全二叉搜索树的图就明白了

# -*- coding: utf-8 -*- 
# @Time : 2019/6/25 10:48 
# @Author : ValarMorghulis 
# @File : 1064.py
from math import log

ans = list()
a=list()

def level(s, e, index):
    if s > e:
        return
    global ans
    tot = e - s + 1
    l = int(log(tot + 1) / log(2))
    leaves = tot - 2 ** l + 1
    root = s + min(leaves, 2 ** (l - 1)) + 2 ** (l - 1) - 1
    ans[index]=a[root]
    level(s, root - 1, index*2+1)
    level(root + 1, e, index*2+2)


def solve():
    global ans, a
    n = int(input())
    ans = [0 for i in range(n)]
    a = list(map(int, input().split()))
    a.sort()
    level(0, n - 1, 0)
    print(*ans)


if __name__ == "__main__":
    solve()