105. 从前序与中序遍历序列构造二叉树

根据一棵树的前序遍历与中序遍历构造二叉树。

注意:
你可以假设树中没有重复的元素。

例如，给出

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

本题是一道比较复杂的题目，如果你没有做过的话。
如果是不是代码的实现，本题也是比较简单的，我们通过前序知道了根节点是3，然后去中序遍历看，在3前面的就是左子树，在3后面的就是右子树，然后继续看前序遍历，9是第一个节点就是左子树的第一个节点，20是右子树的第一个节点，然后又继续中序遍历，可以发现15是20的左子树，7是20的右子树，那本题就解决了。

那如何将上面的思考过程转化为代码呢？
这其实并不算太难，就是有点复杂，我们需要规划一下流程，那就是我们每次创建节点，都是通过前序遍历来创建的节点，我们确定左右子树是通过中序遍历的。
我们可以先从前序遍历中找到本轮我们需要的节点，然后去中序遍历里，找到中序遍历里的这个节点，在这个节点之前的就是左子树，在这个节点之后的就是右子树。

因此代码如下:

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        if (preorder.length == 0 || inorder.length == 0)
            return null;
        return build(preorder,0,preorder.length-1,inorder,0,inorder.length-1);
    }

    TreeNode build(int[] preorder,int preStart,int preEnd,int[] inorder,int inStart,int inEnd){

        TreeNode root = new TreeNode(preorder[preStart]);
        root.left = null;
        root.right = null;

        int inRoot = inStart;
        while (inRoot<=inEnd && inorder[inRoot] != preorder[preStart]) inRoot++;
        int leftlen = inRoot-inStart;
        int rightlen = inEnd - inRoot;
        if (leftlen != 0)
        root.left = build(preorder,preStart+1,preStart+leftlen,inorder,inStart,leftlen+inStart-1);
        if (rightlen != 0)
        root.right = build(preorder,preStart+leftlen+1,preStart+leftlen+rightlen,inorder,inRoot+1,inEnd);
        return root;

    }
}

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。