PTA Tree Traversals

最新推荐文章于 2021-08-12 16:19:56 发布

原创最新推荐文章于 2021-08-12 16:19:56 发布 · 208 阅读

0 ·

CC 4.0 BY-SA版权

树专栏收录该内容

10 篇文章

订阅专栏

本文介绍了一种算法，该算法接收一棵树的后序遍历和中序遍历序列，通过递归方式确定根节点及左右子树，最终输出树的层序遍历结果。代码实现采用Python，详细解析了如何从给定的遍历序列中重构树结构并进行层序遍历。

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

链接

https://pintia.cn/problem-sets/994805342720868352/problems/994805485033603072

题解

这题就是告诉你后序遍历和中序遍历，让你输出层序遍历（就是像 $B F S$ 那样按深度递增便利）
根据后序遍历，我可以直接确定根，然后在中序遍历的序列中找到根，根的左侧就是左子树，右侧就是右子树，左右子树在遍历序列中所占的区间也是一样的范围
递归下去

代码

def work(list1,list2,deepth):
    root = list1[-1]
    c[deepth].append(root)
    mid = list2.index(root)
    if mid!=0:
        work(list1[:mid],list2[:mid],deepth+1)
    if mid!=len(list2)-1:
        work(list1[mid:-1],list2[mid+1:],deepth+1)

n = int(input())
postorder = [int(x) for x in input().split()]
inorder = [int(x) for x in input().split()]
c=[]
for i in range(n+10): c.append([])
work(postorder,inorder,0)
i=0
while(len(c[i+1])>0):
    print(*c[i],end=' ')
    i=i+1
print(*c[i])