剑指 Offer 07. 重建二叉树

这是我参与8月更文挑战的第3天，活动详情查看：8月更文挑战”

剑指 Offer 07. 重建二叉树

输入某二叉树的前序遍历和中序遍历的结果，请构建该二叉树并返回其根节点。

假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

示例 1:

Input: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7] Output: [3,9,20,null,null,15,7] 示例 2:

Input: preorder = [-1], inorder = [-1] Output: [-1]

解题思路

1. 前序遍历数组：根节点+左子树+右子树
2. 中序遍历数组：左子树+根节点+右子树
3. 前序遍历的第一个节点就是根节点，在中序遍历的数组中找出根节点的位置，根节点的左边为左子树，右边为右子树，因此我们可以得出左右子树节点的数量，因此我们也可以根据这个划分出前序遍历中，左右子树的区间。 只要我们在中序遍历中定位到根节点，那么我们就可以分别知道左子树和右子树中的节点数目。由于同一颗子树的前序遍历和中序遍历的长度显然是相同的，因此我们就可以对应到前序遍历的结果中，对左右子树进行定位。

因此，主要思路就是用前序数组的第一个元素为切割点，先切中序数组，根据中序数组切割的结果获得左右子树的长度，反过来在切前序序数组。

代码

/**
 * 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) {
          return build(preorder,0,preorder.length-1,inorder,0,inorder.length-1);
    }
    public TreeNode build(int[] preorder,int pl,int pr,int[] inorder,int il,int ir) {

        if (pl>pr) return null;
        
        //
        TreeNode root = new TreeNode(preorder[pl]);
        if (pl==pr) return new TreeNode(preorder[pl]);
        int i=il;
        for (;i<=ir;i++)
        {
            if (root.val==inorder[i])
                break;

        }
        root.left=build(preorder, pl+1, pl+i-il, inorder, il, i-1);
        root.right=build(preorder, pr-(ir-i-1), pr, inorder, i+1, ir);
        return root;

    }
}