剑指 Offer 07 重建二叉树java题解

╥﹏╥
我数据结构学的是真垃圾。。
https://leetcode-cn.com/problems/zhong-jian-er-cha-shu-lcof/solution/mian-shi-ti-07-zhong-jian-er-cha-shu-di-gui-fa-qin/

分析

1.前序序列中的根节点索引
2.根据根节点，在中序序列中，划分为左子树右子树，知道了左右子树的起始位置。
3.根据根节点，左子树右子树的长度，求出左子树的根节点和右子树的根节点。
4.对左子树和右子树继续递归。

代码

class Solution {
    int[] po;
    HashMap<Integer,Integer> dic=new HashMap<>();
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        po=preorder;
        //为了找到先序序列的数在中序序列中的位置
        for(int i=0;i<inorder.length;i++){
            dic.put(inorder[i],i);
        }
        //根节点在先序序列索引，中序序列左边界，中序序列右边界
        return recur(0,0,inorder.length-1);
    }
    //left和right是针对中序序列而言的
    public TreeNode recur(int pre_root,int inleft,int inright){
    	//没有子树返回null
        if(inleft>inright) return null;
        TreeNode root=new TreeNode(po[pre_root]);
        //找到上一个根节点在中序序列中的索引
        int i=dic.get(po[pre_root]);
        //左子树根节点在先序序列的索引：上一个根节点索引+1
        root.left=recur(pre_root+1,inleft,i-1);
        //右子树根节点在先序序列的索引：上一个根节点索引+左子树长度+1
        root.right=recur(pre_root+i-inleft+1,i+1,inright);
        return root;
    }
}

结果

复杂度

时间O(n):遍历建立HashMap O(n)。每次搜索建立新节点O(1),共N个节点，因此递归占用O(n)
空间O(n):HashMap占O(n),递归操作中系统需使用 O(N)额外空间