9度OC题目号:1385
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并输出它的后序遍历序列。
由于前序遍历的第一个数字就是该树的根,记录该根,中序遍历结果中根节点所在位置左边全部的数字为其(即根节点)左子树的中序遍历,右边全部的数字为其(根节点)右子树的中序遍历。
递归该过程就可重建二叉树。
比如例子中,先取1为根节点,然后在中序遍历结果中1左边的472为根节点的左子树的中序遍历结果,5386为根节点的右子树的中序遍历结果。递归考虑左子树和右子树,全树根节点的左右节点分别为2,3。此时已经重建了3个节点,由此递归一步步重建整个树。
简单起见没有使用泛型且直接在主类中输入
public class Main{
private static boolean isError;
public static void main(String[] args) {
isError = false;
int t = 8;
int[] a = {1,2,4,7,3,5,6,8};
int[] b = {4,7,2,1,5,3,8,6};
Node n = buildTree(a, b, 0, t-1, 0, t-1);
if (isError) {
System.out.println("No");
}else {
postOrder(n);
System.out.println();
}
}
private static Node buildTree(int[] a, int[] b,int as,int ae, int bs, int be) {
if (isError || as > ae || bs > be) {
return null;
}
Node node = new Node(a[as]);
//找出树顶点在 中序数组 中的位置
int i = bs;
for (; i <= be; i++) {
if (b[i] == a[as]) {
break;
}
}
if (i == be + 1) {
isError = true;
return null;
}
int temp = i - bs; //左子树长度
node.lNext = buildTree(a, b, as + 1, as + temp, bs, i-1);
node.rNext = buildTree(a, b, as + 1 + temp, ae, i + 1, be);
return node;
}
private static void postOrder(Node node){
if(node==null||isError)
return;
postOrder(node.lNext);
postOrder(node.rNext);
System.out.print(node.data+" ");
}
public static class Node{
private Node lNext;
private Node rNext;
private int data;
public Node(){}
public Node(int data) {
this.data = data;
}
}
}