参考CN-Notes:https://cyc2018.github.io/CS-Notes/#/
题目描述:
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
思路:
这道题主要是要理解中序遍历和前序遍历的顺序问题。我们通过前序遍历可以知道前序遍历的第一个节点一定是根节点,然后我们找出中序遍历中根节点的位置就知道左子树和右子树的位置,然后依次类推,直到没有左右子树。
代码如下:
/**
* Definition for binary tree
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
import java.util.*;
public class Solution {
//保存中序遍历中元素的索引,便于查找到根节点在中序遍历中的位置,而不用一次次地去遍历
private Map<Integer,Integer> indexForInOrders =
new HashMap<>();
public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
for(int i=0;i<in.length;i++){
indexForInOrders.put(in[i],i);
}
return reConstructBinaryTree(pre,0,pre.length-1,0);
}
private TreeNode reConstructBinaryTree(int[] pre,int preL,int preR, int inL){
//验证索引是否合理
if(preL > preR){
return null;
}
//以当前谦虚遍历的第一个节点为根节点
TreeNode root = new TreeNode(pre[preL]);
//获取到根节点在中序遍历中的位置
int inIndex = indexForInOrders.get(root.val);
//获取左子树的位置
int leftTreeSize = inIndex - inL;
//同理构造根节点的左子树和右子树
root.left = reConstructBinaryTree(pre,preL+1,preL+leftTreeSize,inL);
root.right = reConstructBinaryTree(pre,preL+leftTreeSize+1,preR,inL+leftTreeSize+1);
return root;
}
}
每次我自己写都写不出来,呜呜呜,只能强记一波了。