剑指 Offer 07. 重建二叉树--dfs

一、题目

二、方法

如果是计算题的话，感觉可以自己画出来，但是写代码不太会...主要还是递归的思路没想明白。还可以用栈来做，不过栈目前还不是很熟悉，先递归写一个版本。

思路：

1.从前序遍历里面拿到根节点。

2.从中序遍历里面找到这个根节点的位置idx.[可以用哈希表存中序遍历的值和id]

3.根节点的左边 --->idx - 1 就是左子树范围（从而可以得到左子树的元素个数left_num，然后再从前序遍历中根节点后面移动left_num区间大小就可以递归了）， idx + 1 --> right 就是右子树范围（右子树同理。）

4.注意终止条件

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    Map<Integer, Integer>map = new HashMap();
    int[]preorder;// 保留的先序遍历
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        // dfs 或者 用栈模拟
        // 前序遍历：根 - 左 - 右
        // 中序遍历：左 - 根 - 右
        this.preorder = preorder;
        // map 便于快速找到中序遍历中 根节点的位置
        for(int i = 0 ;  i < preorder.length ; i++){
            map.put(inorder[i], i);
        }
        // 需要返回一棵树
        return dfs(0, 0, preorder.length - 1);
    }
// 注意：本文方法只适用于 “无重复节点值” 的二叉树。
// root_id -- 先序遍历的根节点索引
// left_id -- 中序遍历的左子树索引
// right_id --中序遍历的右子树索引
    public TreeNode dfs(int root_id, int left_id, int right_id){
        if(left_id > right_id) return null;
        TreeNode node = new TreeNode(preorder[root_id]); // 赋值根节点
        int idx = map.get(preorder[root_id]); // 找到中序遍历中根节点的位置
        node.left = dfs(root_id + 1, left_id, idx - 1); // 更新左子树
        node.right = dfs(root_id + (idx - left_id) + 1, idx + 1, right_id); // 更新右子树
        return node;

    }
}

三、结果