JAVA L2-006

 概要

作为一道l2的题目，只要40行就可以写完可算是相当简单了

前提是没有用二叉树去做这道题

代码思想：

1、使用TreeMap储存层序遍历的结果

2、使用Vector（动态数组储存输入的后序及中序）其实理论上用ArrayList即可，Vector有个特点叫做线程安全，但会造成性能损失

---

代码

import java.util.Scanner;
import java.util.TreeMap;
import java.util.Vector;

public class Main{
	private static Vector<Integer> post = new Vector<Integer>();
	private static Vector<Integer> in = new Vector<Integer>();
	private static TreeMap<Integer, Integer> level = new TreeMap<Integer, Integer>();
	//储存在类中作为全局变量，可以随意调用，其方便性在下面会体现
	public static void main(String[] args) {
		Scanner sc =new Scanner(System.in);
		int n = sc.nextInt();
		for (int i = 0; i < n; i++) {
			post.add(sc.nextInt());
		}
		for (int i = 0; i < n; i++) {
			in.add(sc.nextInt());
		}
		restoreTree(n-1, 0, n-1, 0);
        //以上都是题目要求进行的输入部分
		//以下我们要做的就是重建二叉树，然后输出

		int pos = 0;
		for(int i : level.keySet()) {
			if(pos++ != 0 ) {
				System.out.print(" ");
			}
			System.out.print(level.get(i));
		}

        /*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());
        }*/
		
		/*for(Map.Entry<Integer, Integer>entry : level.entrySet()) {
			System.out.print(entry.getValue()+(entry.getKey()!=n-1?" ":""));
		}*/
	}
	private static void restoreTree(int post_end, int mid_start,int mid_end, int index) {
		if(mid_start > mid_end)return;                          //说明数组为空，直接返回；
		int i = mid_start;
		while(i<mid_end && in.get(i)!=post.get(post_end)) {
			i++;
		}
		level.put(index, post.get(post_end));
		restoreTree(post_end-1-mid_end+i, mid_start, i-1, 2*index+1);//进行递归
		restoreTree(post_end-1, i+1, mid_end, 2*index+2);
	}
}

---

理解

关于这个restoreTree方法，它的原理是通过二叉树后序遍历以及中序遍历对每个元素进行定位

while(i<mid_end && in.get(i)!=post.get(post_end)) {
            i++;
        }

这一句就是这个定位方法的最关键部分

以下是写出其方法部分步骤的图解。