【剑指Offer】面试招聘题目2：重建二叉树

题目链接：http://ac.jobdu.com/problem.php?cid=1039&pid=1

题目描述：

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

参考网址：http://zephiruswt.blog.51cto.com/5193151/885951

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;
    }

}