剑指Offer---面试题6:重建二叉树

题目:输入某二叉树的连续遍历和后序遍历结果,请重建此二叉树.输出它的头结点.(假设输入的前序遍历和后序遍历结果中都不含重复数字)

---

分析:

• 前序遍历的第一个数字总是树的根节点的值,假设这个值为A;

• 在中序序列中A位于p位置处,A左边的节点,都是A的左子树的中序遍历,A右边的节点都是A的右子数的中序遍历;

• 另一方面,前序序列中第二个位置到p位置,也是A左子树的前序遍历,剩余的就是A右子树的前序遍历;

• 假设A左边的节点分别为B,C,D,E,F右边的节点分别为G,H,I,J,K,L

• 那么B,C,D,E,F和G,H,I,J,K,L又可以分别看做两棵树,再分别到前序序

  列中去找到根节点,以及他们的左右子树;

• 以此类推,形成了一个递归的结构;

• 递归的出口是:子树前序序列的第一个数的索引,大于子树前序序列的最后一个数的索引; 或者 子树中序序列的第一个数的索引,大于子树中序序列的最后一个数的索引.

---

代码:

public class Solution {
    public TreeNode reConstructBinaryTree(int[] pre, int[] in) {
        TreeNode root = reConstructBinaryTree(pre, 0, pre.length - 1, in, 0,
                in.length - 1);
        return root;
    }

    /**
     * 构造出二叉树的方法
     * 
     * @param pre
     *            子树的前序遍历数组
     * @param startPre
     *            子树的前序遍历数组第一个数在pre数组中的索引
     * @param endPre
     *            子树前序遍历数组最后一个数在pre数组中的索引
     * @param in
     *            子树的中序遍历数组
     * @param startIn
     *            子树的中序遍历数组第一个数在pre数组中的索引
     * @param endIn
     *            子树中序遍历数组最后一个数在pre数组中的索引
     * @return root 根节点
     */
    private TreeNode reConstructBinaryTree(int[] pre, int startPre, int endPre,
            int[] in, int startIn, int endIn) {

        if (startPre > endPre || startIn > endIn)
            return null;
        TreeNode root = new TreeNode(pre[startPre]);

        for (int i = startIn; i <= endIn; i++)
            if (in[i] == pre[startPre]) {
                root.left = reConstructBinaryTree(pre, startPre + 1, startPre + i - startIn, in, startIn, i - 1);
                root.right = reConstructBinaryTree(pre, startPre+(i-startIn)+1, endPre, in, i + 1, endIn);
            }

        return root;
    }
}

难点:递归函数的参数是本题目的一个难点,要分析出

root.left = reConstructBinaryTree(pre, startPre + 1, startPre + i - startIn, in, startIn, i - 1);
root.right = reConstructBinaryTree(pre, startPre+(i-startIn)+1, endPre, in, i + 1, endIn);
            }

确实不是一件容易的事情.