[LeetCode]（面试题07）重建二叉树

题目

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

例如，给出

前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]

返回如下的二叉树：

    3
   / \
  9  20
    /  \
   15   7

限制：

0 <= 节点个数 <= 5000

解题思路

详细思路请参考 105. 从前序与中序遍历序列构造二叉树

代码

/**
 * 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) {
        int n = inorder.length;
        Map<Integer, Integer> map = new HashMap<Integer, Integer>();
        // 在哈希表中存入中序遍历时的根节点索引
        for(int i=0; i<n; i++){
            map.put(inorder[i], i);
        }
        return buildTree(preorder, inorder, 0, n-1, 0, n-1, map);
    }

    public TreeNode buildTree(int[] preorder, int[] inorder, int preleft, int preright, int inleft, int inright, Map<Integer, Integer> map){
        // 若左边索引大于右边索引，表示序列为空
        if(preleft > preright || inleft > inright){
            return null;
        }
        // 前序遍历第一个节点为根节点
        TreeNode root = new TreeNode(preorder[preleft]);
        int rootId = map.get(preorder[preleft]);
        // 左子树长度为 rootId - inleft
        // 右子树长度为 inright - rootId
        root.left = buildTree(preorder, inorder, preleft+1, preleft + rootId - inleft, inleft, rootId-1, map);
        root.right = buildTree(preorder, inorder, preleft + rootId - inleft + 1, preright, rootId+1, inright, map);
        return root;
    }
}