给定一棵二叉树的中序遍历和前序遍历,请你先将树做个镜面反转,再输出反转后的层序遍历的序列。所谓镜面反转,是指将所有非叶结点的左右孩子对换。这里假设键值都是互不相等的正整数。
输入格式:
输入第一行给出一个正整数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);//右子树
}
}