力扣之递归——重建二叉树
这道题的思路就是清楚前续遍历与中序遍历之间的联系,怎样将两者联系起来,其实里面会有些规律,其实也不是规律,更多的前续遍历与中序遍历的实质。
思路
使用指针解决
这道题只需要使用三个指针就可以了,分别是前序遍历中使用到的指针pre_start,它代表前续遍历开始的位置;还有两个中序遍历中使用的指针,分别为inStart,inEnd,代表着中序遍历开始和结束的位置,其实就是对中序遍历进行拆解。下面就用一张图来进行分析一下。
他的前序遍历是:[3,9,8,5,2,20,15,7]
他的中序遍历是:[5,8,9,2,3,15,20,7]
这里的思路就是把前序遍历中正在遍历的节点的位置在中序遍历的数组中能找到相应的位置,如果以index为前续遍历中的某个值在作为中序遍历的索引,如果以index为根节点进行划分的话,那么中序遍历中
[0,index-1]就是根节点左子树的所有节点
[index+1,inorder.length-1]就是根节点右子树的所有节点
中序遍历相对来说简单划分,前序遍历稍微麻烦一点
左子树的话就是:preStart+1;
右子树的思路:就是当前节点的位置加上左子树的数量+1就是右子树开始的位置(preStart+index-inStart+1)
下面来上代码
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
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);
}
public TreeNode recur(int preStart,int inStart,int inEnd){
if(inStart>inEnd){
return null;
}
TreeNode root=new TreeNode(po[preStart]);
int i=dic.get(po[preStart]);
root.left=recur(preStart+1,inStart,i-1);
root.right=recur(preStart+i-inStart+1,i+1,inEnd);
return root;
}
}