题目链接:
题目描述:
题目描述: 输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{ 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”。样例输入: 81 2 4 7 3 5 6 84 7 2 1 5 3 8 681 2 4 7 3 5 6 84 1 2 7 5 3 8 6样例输出: 7 4 2 5 8 6 3 1 No
参考网址:
Java代码,测试通过,但是OJ测试,Runtime Error
import java.util.Scanner;/** * 题目二:重建二叉树 * * @author yinger */public class NineOj_t2 { static int count = 0; public static void main(String[] args) { Scanner cin = new Scanner(System.in); int n; while (cin.hasNextInt()) { count =0; n = cin.nextInt(); int[] pre = new int[n]; int[] ino = new int[n]; for (int i = 0; i < n; i++) { pre[i] = cin.nextInt(); } for (int i = 0; i < n; i++) { ino[i] = cin.nextInt(); } Node root = new Node(pre[0]);// 二叉树的根节 buildTree(root, pre, 0, n - 1, ino, 0, n - 1); StringBuffer buffer = new StringBuffer(); postOrder(root,buffer);// 后序遍历 if (count == n) { //肯定只会append n次,如果不是,那么二叉树是错误的 System.out.println(buffer.toString()); }else{ System.out.println("No"); }// if (buildTree(root, pre, 0, n - 1, ino, 0, n - 1)) { // 重建二叉树// postOrder(root);// 后序遍历// } else { // 不存在这样的二叉树// System.out.println("No");// } } } // 后序遍历二叉树 private static void postOrder(Node root, StringBuffer buffer) { if (root == null) { return; } postOrder(root.left, buffer); postOrder(root.right, buffer); visit(root,buffer); } private static void visit(Node root, StringBuffer buffer) {// System.out.print(root.value + " "); buffer.append(root.value+" "); count ++; } // 构建二叉树 private static void buildTree(Node root, int[] pre, int pre_start, int pre_end, int[] ino, int ino_start, int ino_end) { if (pre_start > pre_end || pre_end > pre.length) { // pre起点和终点的索引错了,或者超出了范围 return; }// if (pre_start == pre_end && ino_start == ino_end) { // 在只有一个元素的时候要检查数字是否相同!不同的话就不能构成二叉树// return pre[pre_start] == ino[ino_start];// } // boolean flag_left = true,flag_right=true;s int index = findRootIndex(ino, root.value);// 查找root在ino中的索引 int len_left = index - ino_start;// root的左子树的长度 int len_right = ino_end - index;// root的右子树的长度 if (len_left > 0) { // 存在左子树 Node leftRoot = new Node(pre[pre_start + 1]);// 左根节点 root.left = leftRoot; /* flag_left = */buildTree(leftRoot, pre, pre_start + 1, pre_start + len_left - 1, ino, ino_start, index - 1); } // if (!flag_left) { //如果左边的子树出现了问题,那么就返回false,表示二叉树不存在 // return false; // } if (len_right > 0) { // 存在右子树 Node rightRoot = new Node(pre[pre_start + len_left + 1]);// 右根节点 root.right = rightRoot; /* flag_right = */buildTree(rightRoot, pre, pre_start + len_left + 1, pre_start + len_left + len_right - 1, ino, index + 1, ino_end); }// return flag_right; } // 在中序序列中查找根节点的索引 private static int findRootIndex(int[] ino, int value) { for (int i = 0; i < ino.length; i++) { if (ino[i] == value) { return i; } } return -1; }}class Node { int value; Node left; Node right; public Node(int value) { this.value = value; }}