(蓝桥杯)试题 算法训练 求先序排列

Posted 2021-09-19 nuist__NJUPT

试题 算法训练 求先序排列

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

样例输入
　　BADC
　　BDCA
样例输出
ABCD

import java.util.Scanner;

public class Main{
    public static void preOrder(String s1, String s2, int l1, int r1, int l2, int r2){
        int k = -1 ;
        for(int i=l1; i<r1; i++){
            if(s1.charAt(i) == s2.charAt(r2-1)){ // 中序的元素等于先序的最后一个元素，则该元素为根节点
                System.out.print(s1.charAt(i)) ;
                 k = i ;
                break ;
            }
        }
        if(k > l1){ //有左子树
            preOrder(s1,s2,l1,k,l2,k-l1+l2) ;
        }
        if(r1 > k+1){ //有右子树
            preOrder(s1,s2,k+1,r1,k-l1+l2,r2-1) ;
        }
    }
    public static void main(String[] args){
        Scanner input = new Scanner(System.in) ;
        String s1 = input.next() ;
        String s2 = input.next() ;
        preOrder(s1,s2,0,s1.length(),0,s2.length());
    }
}