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

题目描述

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