5-剑指 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]

限制：0 <= 节点个数 <= 5000

---

思路

1.遍历前序数组，找到当前树的根节点。

2.拿这个根节点，去中序遍历中找到该元素对应的索引下标，根据索引，将中序遍历分为两部分：左区间为左树节点，右区间为右树节点。

3.将左树连到root.left，右树连到root.right，重复上述过程。

变量定义：

index:表示当前处理到前序遍历的哪个节点。

[left, right):表示index对应的根节点在中序遍历的结果集。

若left >= right:是空树。

若index >= preOrder.length,前序遍历已经结束，构建节点。

如：preorder:[3,9,20,15,7]      inorder:[9,3,15,20,7]

当index=0时，根节点为3，[left, right)为[0,5）

当index=1时，根节点为9，[left, right)为[0,1)

当index=2时，根节点为20，[left, right)为[2,5)

---

代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */

public class Solution {
    //当前处理到前序遍历的哪个位置
    int index = 0;
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        return buildTreeInternal(preorder,inorder,0,inorder.length);
    }

    /**
     * 根据传入的前序遍历和中序遍历的部分区间还原二叉树，返回root
     * [left, right)表示当前节点中序遍历结果
     * 左区间连接到根节点左侧
     * 右区间连接到根节点右侧
     * @param preOrder 前序遍历结果
     * @param inOrder 中序遍历结果
     * @param left
     * @param right
     * @return
     */
    public TreeNode buildTreeInternal(int[] preOrder,int[] inOrder,int left,int right){
        //边界
        if(left >= right){
            //空区间，说明此时中序遍历结果为空，空树
            return null;
        }
        if(index >= preOrder.length){
            //所有节点已经全部处理完毕（前序遍历已经结束）
            return null;
        }
            
        //此时前序结果还有没处理完的元素，从index位置取得一个元素构建根节点
        TreeNode root = new TreeNode(preOrder[index]);
        //继续处理下一个节点
        index++;
        //拿着这个root.val去中序遍历中找到该值对应的索引POS
        int pos = find(inOrder,left,right,root.val);
        //处理左树
        root.left= buildTreeInternal(preOrder,inOrder,left,pos);
        //处理右树
        root.right = buildTreeInternal(preOrder,inOrder,pos + 1,right);
        return root;
    }

    /**
     * 在当前中序遍历中寻找root.val对应的索引值
     * @param inOrder
     * @param left
     * @param right
     * @param val
     * @return
     */
    private int find(int[] inOrder,int left,int right,int val){
        for(int i = left; i < right; i++){
            if(inOrder[i] == val){
                return i;
            }
        }
        return -1;
    }
}