╥﹏╥
我数据结构学的是真垃圾。。
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)额外空间