1385:重建二叉树 @jobdu

来源:互联网 发布:淘宝改差评收费 编辑:程序博客网 时间:2024/06/05 15:19

思路很清晰,但是很容易就写错,特别是在写递归的参数时!为了避免混淆,我新建了多个临时变量来表示,这样会看的清楚一些,免得绕进去了!


题目1385:重建二叉树

时间限制:1 秒

内存限制:32 兆

特殊判题:

提交:1574

解决:517

题目描述:

输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{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



重点是这幅图!要弄清楚index的变化关系!



import java.io.BufferedInputStream;import java.io.FileInputStream;import java.io.FileNotFoundException;import java.util.ArrayList;import java.util.List;import java.util.Scanner;public class S6 {public static void main(String[] args) throws FileNotFoundException {BufferedInputStream in = new BufferedInputStream(new FileInputStream("S6.in"));        System.setIn(in);Scanner cin = new Scanner(System.in);while (cin.hasNextInt()) {List<Integer> preorder = new ArrayList<Integer>();List<Integer> inorder = new ArrayList<Integer>();int size = cin.nextInt();for(int i=0; i<size; i++){preorder.add(cin.nextInt());}for(int i=0; i<size; i++){inorder.add(cin.nextInt());}//System.out.println(preorder);//System.out.println(inorder);TreeNode root = rebuild(preorder, inorder, 0, preorder.size()-1, 0, inorder.size()-1);List<Integer> postorderList = new ArrayList<Integer>();postorder(root, postorderList);if(postorderList.size() == preorder.size()){for (Integer integer : postorderList) {System.out.print(integer + " ");}System.out.println();}else{System.out.println("No");}//inorder(root);}}public static TreeNode rebuild(List<Integer> preorder, List<Integer> inorder, int preorderStart, int preorderEnd, int inorderStart, int inorderEnd){int rootVal = preorder.get(preorderStart);int rootInorder = -1;for(int i=inorderStart; i<=inorderEnd; i++){if(inorder.get(i) == rootVal){rootInorder = i;break;}}if(rootInorder == -1){return null;}TreeNode root = new TreeNode(rootVal);int leftSize = rootInorder - inorderStart;int leftPreorderStart = preorderStart+1;int leftPreorderEnd = preorderStart + leftSize;int leftInorderStart = inorderStart;int leftInorderEnd = rootInorder-1;if(leftSize > 0){root.left = rebuild(preorder, inorder, leftPreorderStart, leftPreorderEnd, leftInorderStart, leftInorderEnd);}int rightPreorderStart = leftPreorderEnd+1;int rightPreorderEnd = preorderEnd;int rightInorderstart = rootInorder+1;int rightInorderEnd = inorderEnd;if(leftSize < preorderEnd-preorderStart){root.right = rebuild(preorder, inorder, rightPreorderStart, rightPreorderEnd, rightInorderstart, rightInorderEnd);}return root;}public static void inorder(TreeNode root){if(root != null){inorder(root.left);System.out.print(root.val + " ");inorder(root.right);}}public static void postorder(TreeNode root, List<Integer> postorderList){if(root != null){postorder(root.left, postorderList);postorder(root.right, postorderList);postorderList.add(root.val);}}static class TreeNode{int val;TreeNode left;TreeNode right;public TreeNode(int val){this.val = val;}}}





0 0
原创粉丝点击