【蓝桥杯ALGO-20】求先序排列 Java版

试题 算法训练 求先序排列

资源限制
时间限制：1.0s 内存限制：256.0MB
问题描述
　　给出一棵二叉树的中序与后序排列。求出它的先序排列。（约定树结点用不同的大写字母表示，长度<=8）。
输入格式
　　两行，每行一个字符串，分别表示中序和后序排列
输出格式
　　一个字符串，表示所求先序排列

样例输入
　　BADC
　　BDCA
样例输出
ABCD

Java 代码：

import java.io.*;

public class Main {

    // 用in，post，pre数组分别表示中序，后序和先序
    private static char[] in, post, pre;

    public static void main(String[] args) throws Exception {
        BufferedReader input = new BufferedReader(new InputStreamReader(System.in));
        in = input.readLine().toCharArray();
        post = input.readLine().toCharArray();
        pre = new char[in.length];
        findPre(0, in.length - 1, post.length - 1, 0);
        for (int i = 0; i < pre.length; i++) {
            System.out.print(pre[i]);
        }

    }

    // inStart表示中序的起始位置，inEnd表示中序的终止位置，postIndex表示后序中根节点，preIndex表示要填充的先序的位置
    // 在inStart到inEnd范围内找postIndex位置的根节点，分别求得左子树和右子树的节点个数，确定左子树和右子树在post中的位置，递归求解左子树和右子树
    private static void findPre(int inStart, int inEnd, int postIndex, int preIndex) {
        if (inStart > inEnd)
            return;
        for (int i = inStart; i <= inEnd; i++) {
            if (in[i] == post[postIndex]) {
                pre[preIndex] = in[i];
                // 遍历左子树，后序排列中前i-inStart个节点是当前根节点的左子树的节点
                findPre(inStart, i - 1, postIndex - (inEnd - i) - 1, preIndex + 1);
                // 遍历右子树，后序排列中i+1到根节点下标-1是当前根节点的右子树的节点，preIndex需要跳过左子树的节点
                findPre(i + 1, inEnd, postIndex - 1, preIndex + i - inStart + 1);
                return;
            }
        }
    }
}