题目
如果已知二叉树前序遍历为:1 2 4 7 3 5 6 8,中序遍历为:4 7 2 1 5 3 8 6,求此二叉树的后序遍历。
import java.util.ArrayList;
import java.util.Scanner;
public class Main
{
public static void main(String[] args)
{
Scanner sca = new Scanner(System.in);
// while(sca.hasNext())
// {
int n = sca.nextInt();
ArrayList<Integer> PreOrder = new ArrayList<Integer>();
ArrayList<Integer> InOrder = new ArrayList<Integer>();
for(int i = 0; i < n; i++)
{
PreOrder.add(sca.nextInt());
}
for(int i = 0; i < n; i++)
{
InOrder.add(sca.nextInt());
}
get(n,PreOrder,InOrder);
// }
sca.close();
}
public static void get(int n, ArrayList<Integer> PreOrder, ArrayList<Integer> InOrder)
{
if(PreOrder.size() == 0)
{
return;
}
else if(PreOrder.size() == 1)
{
System.out.print(InOrder.get(0) + " ");
return;
}
ArrayList<Integer> pre1 = new ArrayList<Integer>();
ArrayList<Integer> pre2 = new ArrayList<Integer>();
ArrayList<Integer> in1 = new ArrayList<Integer>();
ArrayList<Integer> in2 = new ArrayList<Integer>();
int i = 0;
for(i = 0; PreOrder.get(0) != InOrder.get(i); i++)
{
in1.add(InOrder.get(i));
pre1.add(PreOrder.get(i+1));
}
get(n,pre1,in1);
int tmp = InOrder.get(i);
for(i++; i < PreOrder.size(); i++)
{
in2.add(InOrder.get(i));
pre2.add(PreOrder.get(i));
}
get(n,pre2,in2);
System.out.print(tmp + " ");
}
}
这道题是今天华为测试竞赛中的题,原题是这样的:
重建二叉树 | |
描述: | 输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并输出它的后序遍历序列。
|
题目类别: | 树 |
难度: | 中级 |
分数: | |
运行时间限制: | 无限制 |
内存限制: | 无限制 |
阶段: | 招聘管理 |
输入: | 输入可能包含多个测试样例,对于每个测试案例, 输入的第一行为一个整数n(1<=n<=1000):代表二叉树的节点个数。 输入的第二行包括n个整数(其中每个元素a的范围为(1<=a<=1000)):代表二叉树的前序遍历序列。 输入的第三行包括n个整数(其中每个元素a的范围为(1<=a<=1000)):代表二叉树的中序遍历序列。
|
输出: | 对应每个测试案例,输出一行: 如果题目中所给的前序和中序遍历序列能构成一棵二叉树,则输出n个整数,代表二叉树的后序遍历序列,每个元素后面都有空格。 如果题目中所给的前序和中序遍历序列不能构成一棵二叉树,则输出”No”。
|
样例输入: | 8 1 2 4 7 3 5 6 8 4 7 2 1 5 3 8 6 8 1 2 4 7 3 5 6 8 4 1 2 7 5 3 8 6 |
样例输出: | 7 4 2 5 8 6 3 1 No |
答案提示: | |
我做了很久,求后序遍历比较简单,只是我不知道题上说的输入完成是以什么为结束标志的,所以在、纠结了很久,然后网断了,然后就账号过期了,我也就不想再想了,只是把核心算法写出来,如果各位知道怎么做的话,欢迎指导。
我写的没考虑输入的遍历能否构成二叉树,所以还是需要完善的。