思路:
需要先将中序数组的【值-索引】信息进行储存,接着只要找到前序遍历的结点在中序遍历中的位置,就可以将中序数组分解为左右子树两部分;
根节点在中序数组中的位置为index,进而可求得子树上的结点数 = index - VinS;
创建四个变量来定位子树的位置:前序中的PreS和PreE、中序中的VinS和VinE;
对于叶子结点和子树这两种状态需要分情况进行处理;
时间复杂度:O(N),N为树的结点数
import java.util.*;
/**
* Definition for binary tree
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
public class Solution {
public int[] pre;
public int[] vin;
public Map<Integer,Integer> map;
//思路:根据前序数组来定位中序的根结点
public TreeNode reConstructBinaryTree(int [] pre,int [] vin) {
//初始化
this.pre = pre;
this.vin = vin;
map = new HashMap<>();
//存储中序数组的位置信息
for(int i = 0; i < vin.length; i++){
map.put(vin[i],i);
}
return getRoot(0,pre.length - 1, 0, vin.length - 1);
}
//传递的信息:树在两种序列中的位置
public TreeNode getRoot(int PreS, int PreE, int VinS, int VinE){
if(PreS > PreE) return null;
if(VinS > VinE) return null;
//如果是叶子结点
int rootVal = pre[PreS];
if(PreS == PreE) return new TreeNode(rootVal);
//如果是子树,先获取根结点
TreeNode root = new TreeNode(rootVal);
//只要找到前序遍历的结点在中序遍历中的位置,就可以将中序数组分解为左右子树两部分
int index = map.get(rootVal);
//左、右子树(index - VinS :子树的结点个数)
root.left = getRoot(PreS + 1, PreS + index - VinS, VinS, index - 1);
root.right = getRoot(PreS + index - VinS + 1, PreE,index + 1,VinE);
return root;
}
}