剑指offer之重建二叉树

原创不易，求点赞~~

1.题目描述

输入某二叉树的前序遍历和中序遍历的结果，请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

例如，给出

前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树：

2.思路分析

这个题目出自剑指offer，也是一道非常经典的题目了。首先我们要知道二叉树遍历的特性。

那么我们就可以看出来，前序遍历的第一个节点一定是根节点，对应的这个节点在中序遍历中，它在中序遍历位置中前边的所有节点，都是这个根节点的左子树，后边的所有节点都是根节点的右子树。

就以上边的这个二叉树为例：

其中，前序遍历中的第一个节点3就是这个二叉树的根节点。

在中序遍历中，根节点之前的节点9就是它的左子树，根节点之后的节点15 20 7就是它的右子树。

然后采用分治的思想，分别递归地处理左右子树。比如针对右子树15 20 7 。找到这个子树在前序遍历中的排列，为20 15 7。可以知道20是这个子树的根节点，15是它的左子树，7是它的右子树。以此类推即可。

3.代码

题目传入了两个参数，一个是前序遍历结果，一个是中序遍历结果。
因为每次是根据前序遍历找根节点，从中序遍历分割左右子树（这也体现了前序遍历和中序遍历的使命~~~），所以把中序遍历放进一个顺序表中方便操作。

List<Integer> list = new ArrayList<>();
        for(int x : inorder){
            list.add(x);
        }

如果这个顺序表为空，就返回null。
根节点就是前序遍历中的第一个元素。即

TreeNode root = new TreeNode(preOrder[0]);

根节点的位置可以这样求：

int pos = list.indexOf(root.val);

那么左子树就是顺序表中0到pos。
右子树就是顺序表中的pos + 1到list.size()。注意这里都是前闭后开。

class Solution {
    int index = 0;
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        index = 0;
        List<Integer> list = new ArrayList<>();
        for(int x : inorder){
            list.add(x);
        }
        return _buildTree(preorder, list);

    }
    public TreeNode  _buildTree(int[] preorder, List<Integer> list){
      if(list.isEmpty()){
          return null;
      }
        TreeNode root = new TreeNode(preorder[index]);
        index++;
        int pos = list.indexOf(root.val);
        root.left = _buildTree( preorder, list.subList(0, pos));
        root.right = _buildTree( preorder, list.subList(pos + 1, list.size()));
        return root;
    }
}