有一棵二叉树,每个节点由一个大写字母标识(最多26个节点)。
现有两组字母,分别表示后序遍历(左孩子->右孩子->父节点)和中序遍历(左孩子->父节点->右孩子)的结果,请你输出层序遍历的结果。
输入
每个输入文件一行,第一个字符串表示后序遍历结果,第二个字符串表示中序遍历结果。(每串只包含大写字母)
中间用单空格分隔。
输出
输出仅一行,表示层序遍历的结果,结尾换行。
样例输入
CBEFDA CBAEDF
样例输出
ABDCEF
代码(带有详细解释):
package Date3Point21;import java.util.*;/*** 有一棵二叉树,每个节点由一个大写字母标识(最多26个节点)。* 现有两组字母,分别表示后序遍历(左孩子->右孩子->父节点)和中序遍历(左孩子->父节点->右孩子)的结果,请你输出层序遍历的结果。*/
public class MainE {private static String result="";//设置全局结果的返回public static void main(String[] args) {Scanner scanner=new Scanner(System.in);String s1 = scanner.next();//后序String s2 = scanner.next();//中序String[] split1 = s1.split("");//切割String[] split2 = s2.split("");Map<Integer,String> nomalSort=new HashMap<>();//用map集合来存储每个节点在数组中的位置以及节点值myfunction(split1,split2,nomalSort,0);//从中序序列和后续序列得到完全二叉树的数组形式,key对应在数组中的位置,value对应该节点的值bianli(nomalSort,0);//得到map集合存储的完全二叉树的数组形式,进行层次遍历,得到结果System.out.println(result);}/*** 层次遍历* @param nomalSort* @param i 起始节点*/private static void bianli(Map<Integer, String> nomalSort, int i) {Queue<Integer> stack=new LinkedList<>();//用队列来实现层次遍历stack.add(i);while (!stack.isEmpty()){Integer pop = stack.poll();result+=nomalSort.get(pop);if(nomalSort.containsKey(pop*2+1)){stack.add(pop*2+1);}if(nomalSort.containsKey(pop*2+2)){stack.add(pop*2+2);}}}/**** @param split1 后序* @param split2 中序* @param nomalSort 正常* @param index 正常顺序中的数组索引*/private static void myfunction(String[] split1, String[] split2, Map<Integer,String> nomalSort, int index) {if(split1.length==0 || split2.length==0){//如果中序或者后序数组的大小为0,说明该子树为空。return;}if(split1.length==1){//如果中序或者后序数组的大小为1,直接可以赋值,而无需做其他操作。nomalSort.put(index,split1[0]);return;}nomalSort.put(index,split1[split1.length-1]);//将父节点加入map集合String parent=split1[split1.length-1];//得到后序序列的最后一个结点,最后一个节点就是该部分的最顶层的节点for(int i=0;i<split2.length;i++){//循环遍历中序序列得到,根结点所在位置,然后即可找到左右子树if(split2[i].equals(parent)){String[] inLeft = Arrays.copyOfRange(split2, 0, i);//中序遍历的左子树String[] inRight = Arrays.copyOfRange(split2, i+1, split2.length);//中序遍历中的右子树String[] postLeft = Arrays.copyOfRange(split1, 0, i);//后序遍历中的左子树String[] postRight = Arrays.copyOfRange(split1, i, split1.length-1);//后序遍历中的右子树myfunction(postLeft,inLeft,nomalSort,index*2+1);//递归进入下一层(左孩子结点)myfunction(postRight,inRight,nomalSort,index*2+2);//递归进入下一层(右孩子结点)break;}}}
}