《剑指Offer_编程题》详解重建二叉树

题目描述

输入某二叉树的前序遍历和中序遍历的结果，请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}，则重建二叉树并返回。

/**
 * Definition for binary tree  给出的二叉树的数据结构
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */

时间限制：1秒 空间限制：32768K

PS：刚接触算法，小白一枚，对于有些题目是苦苦思索而不得。有时候有思路却无法写出代码，有时候连别人给出的答案也要想半天 :(  。遂将自己的理解记录于此，也给他人提供些解题的参考。

言归正传，要解决本题目，首先起码知道前序遍历，中序遍历吧，下面给出《数据结构》严蔚敏 吴伟民 书中的定义

前序遍历：若二叉树为空，则空操作；否则

1. 访问根节点；
2. 先序遍历左字树；
3. 先序遍历右子树；

中序遍历：若二叉树为空，则空操作；否则

1. 中序遍历左子树；
2. 访问根节点；
3. 中序遍历右子树；

OK，明白了这两个遍历，我们可以直接得出的结论是

1. 二叉树的根节点就是前序遍历中的第一个节点，即 1 是根节点
2. 中序遍历序列中的 4、7、2 是二叉树的左子树，5、3、8、6是二叉树的右子树

     图 1

                                                                                               图 2

顺便给个结果图做比对：

图 3

那么，我们是否能够利用递归思想，每次都将左右两边的子树当作一根新的二叉树处理

1. 找出二叉树的根节点
2. 构建左子树----->找出该左子树的根节点，构建该左字树的左子树，构建该左子树的右子树.....直到序列中没有左子树的序列
3. 构建右子树----->找出该右子树的根节点，构建该右字树的左子树，构建该右子树的右子树.....直到序列中没有右子树的序列

下面给出 欲风 同学的代码，我做了点注释

package com;
public class Solution {

    public class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;

        TreeNode(int x) {
            val = x;
        }
    }
    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 前序遍历序列中 新子树开始的序号的下标
     * @param endPre   前序遍历序列中 新子树结束的序号的下标
     * @param in 中序遍历序列的数组
     * @param startIn  中序遍历序列中 新子树开始的序号的下标
     * @param endIn    中序遍历序列中 新子树结束的序号的下标
     * @return
     */
    private TreeNode reConstructBinaryTree(int [] pre,int startPre,int endPre,int [] in,int startIn,int endIn) {
        /**
         * 作为退出递归的条件
         * 当该子树在序列中的开始的下标大于结束的下标时，说明已经没有子树，将该子树置为null
         */
        if(startPre>endPre||startIn>endIn)
            return null;
        /**
         * 新子树的根节点（一定是当前前序遍历序列中的pre[startPre]）
         */
        TreeNode root=new TreeNode(pre[startPre]);


        for(int i=startIn;i<=endIn;i++)
            /**
            根据前序遍历的序列和中序遍历的序列，找出新子树的根节点，并递归计算新子树下的左子树和右子数
             i即是中序遍历序列中本树的根节点的位置
             */
            if(in[i]==pre[startPre]){
                /**
                 *startPre = startPre + 1 ：startPre是前序遍历序列中本树的根节点，所以下次递归要从本树的子树序列开始
                 * endPre = startPre + i -startIn :下次递归本树的子树序列结束的位置，在中序遍历序列中i的左边的所有序列都是本树的左子树，所以i也是本树的左子树的个数（如图一）
                 */
                root.left=reConstructBinaryTree(pre,startPre+1,startPre+i-startIn,in,startIn,i-1);
                /**
                 * 里面的加加减减好难解释啊...有种只可意会不可言传的感觉。。。
                 * 相信你们将程序单步调试几遍后就会明白
                 */
                root.right=reConstructBinaryTree(pre,i-startIn+startPre+1,endPre,in,i+1,endIn);
                break;
            }
        return root;
    }
}

给出一张递归图解，写了一半左右，举一反三吧 :)