L2-011 玩转二叉树 (25分)java解析

给定一棵二叉树的中序遍历和前序遍历，请你先将树做个镜面反转，再输出反转后的层序遍历的序列。所谓镜面反转，是指将所有非叶结点的左右孩子对换。这里假设键值都是互不相等的正整数。

输入格式：

输入第一行给出一个正整数N（≤30），是二叉树中结点的个数。第二行给出其中序遍历序列。第三行给出其前序遍历序列。数字间以空格分隔。

输出格式：

在一行中输出该树反转后的层序遍历的序列。数字间以1个空格分隔，行首尾不得有多余空格。

输入样例：

7
1 2 3 4 5 6 7
4 1 3 2 6 5 7

输出样例：

4 6 1 7 5 3 2

Note:

 

不需要构建二叉树再进行广度优先搜索，可以加一个变量index,表示二叉树的根节点时index对应的二叉树为0，每一次前序遍历递归时，把根节点的下标及其多对应的值存入level中，镜面反转是将左右子树交换，所以只要改变index的值。左子树为index*2+2,右子树为index*2+1。这样递归完成之后level按照索引大小排序就是层序遍历的顺序.排序之后按照索引大小，右子树排在前面，左子树排在后面，就是镜像之后的顺序。(level的类型是TreeMap,是有序的，按照索引大小进行排序）

代码：

import java.util.*;

public class Main {
    private static Vector<Integer> pre = new Vector<>();
    private static Vector<Integer> in = new Vector<>();
    private static TreeMap<Integer,Integer> level = new TreeMap<>();//有序/LinkedHashMap
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        for(int i = 0; i<n; i++)in.add(sc.nextInt());
        for(int i = 0; i<n; i++)pre.add(sc.nextInt());
        levelorder(0,0,n-1,0);
        Set set = level.entrySet();
        Iterator it = set.iterator();
        Map.Entry entry = (Map.Entry)it.next();
        System.out.print((Integer)entry.getValue());
        while(it.hasNext()){
            entry = (Map.Entry)it.next();
            System.out.print(" "+(Integer)entry.getValue());
        }

    }
    private static void levelorder(int root, int start, int end, int index) {
        if(start > end)return;
        int i = start;
        while(i < end && in.get(i) != pre.get(root))i++;
        level.put(index,pre.get(root));//根
        levelorder(root + 1, start, i - 1, 2 * index + 2);//左子树
        levelorder(root + 1 + i - start, i + 1, end, 2 * index + 1);//右子树
    }
}