ALGO-20 VIP试题 求先序排列 python

问题描述
　　给出一棵二叉树的中序与后序排列。求出它的先序排列。（约定树结点用不同的大写字母表示，长度<=8）。

输入格式

　　两行，每行一个字符串，分别表示中序和后序排列

输出格式

　　一个字符串，表示所求先序排列

　　样例输入
　　BADC
　　BDCA

样例输出

ABCD

思路：后序遍历（左右中）的最后一个节点肯定是树的根节点。找到根节点之后，在中序遍历（左中右）中，根节点的左边的节点都是左子树的节点，右边的点都是右子树的点。这样就可以找到中序遍历的左子树和右子树，根据左子树和右子树可以在后序遍历中找到后序遍历的左子树和右子树。接着使用递归继续分别找左子树和右子树的根节点、左子树、右子树。

由于需要打印先序遍历（中左右）所有在递归过程中，每次找到根节点都先打印出来，然后先递归左子树在递归右子树。

def output(listbetw,listback):
    if len(listbetw)>=1:
        mount = listback[-1]
        n = listbetw.index(mount)
        print(mount,end='')
        left = listbetw[0:n]#左子树的中序遍历
        right = listbetw[n+1:]#右子树的中序遍历
        leftb = listback[0:len(left)]#左子树的后序遍历
        rightb = listback[len(left):-1]#右子树的后序遍历
        output(left,leftb)
        output(right,rightb)

a = list(input())
b = list(input())
output(a,b)