重建二叉树 -- 漫漫算法路 刷题篇

题目描述

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

 * }

 */

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;

    }

    private static 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);   //这里的递归参数设计的是真的巧妙，解释下这个前序的尾元素：startPre+i-startIn  //其实就是算出中序遍历的根节点 i ，减去 startIn ，即为当前根节点的左子树元素个//数，再从前序遍历根节点往后相应的单位内的元素
  就是左子树的全部元素了

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

            }

        }

        return root;

    }

}