【剑指Offer】重建二叉树

剑指Offer-重建二叉树

题目描述

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

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

返回如下的二叉树

    3
   / \
  9  20
    /  \
   15   7

问题分析

前序遍历：根节点->左子树->右子树
中序遍历：左子树->根节点->右子树
首先要找出三个根节点：根结点、左子树根节点、右子树根节点。

• 前序遍历的首元素为根结点node
• 中序遍历中找到根节点node的索引值，node将其分为左子树|根节点|右结点
• 根据中序遍历得到的左子树和右子树数量，可以将前序遍历分为根节点|左子树|右子树
  对于树的左右子树，采用递归分治的思想找到左右子树根节点。

算法

首先，我们定义根节点在前序遍历的索引root，子树在中序遍历中的左边界left和有边界right
递归边界：left>right，终止递归，越过叶结点，返回null

• 建立根节点 node ： 节点值为 preorder[root] ；
• 划分左右子树： 查找根节点在中序遍历 inorder 中的索引 i ；
• 构建左右子树：开启左右子树递归
• 回溯返回：回溯返回node，作为上一个根节点的左右子结点

问题解决

int[] preorder;
    HashMap<Integer, Integer> dic = new HashMap<>();
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        this.preorder = preorder;
        for(int i = 0; i < inorder.length; i++)
            dic.put(inorder[i], i);
        return recur(0, 0, inorder.length - 1);
    }
    TreeNode recur(int root, int left, int right) {
        if(left > right) return null;                          // 递归终止
        TreeNode node = new TreeNode(preorder[root]);          // 建立根节点
        int i = dic.get(preorder[root]);                       // 划分根节点、左子树、右子树
        node.left = recur(root + 1, left, i - 1);              // 开启左子树递归
        node.right = recur(root + i - left + 1, i + 1, right); // 开启右子树递归
        return node;                                           // 回溯返回根节点
    }