题目描述:
根据一棵树的前序遍历与中序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。
例如,给出
前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树:
3
/ \
9 20
/\
15 7
解题思路:
前序遍历:根左右 中序遍历:左根右
我们明显的可以看出先序遍历中,一开始的元素一定是整个二叉树的根节点,从而我们把这个元素找到它在中序元素中的位置,我们就能得到该节点的左子树和右子树,进而我们可以使用分治的思想来处理该问题,主要采用递归的方式,我们再按照同样的操作去分别操作该节点的左子树和右子树,然后操作完后进行回溯,最终连接而成的树就是我们想要的二叉树。
时间复杂度:O(n) 空间复杂度:O(n) n为节点中的个数
代码:
public class LC105 {
private Map<Integer, Integer> ansMap;
public TreeNode buildTree(int[] preorder, int[] inorder) {
int n = preorder.length;
//使用哈希来寻找根节点在中序遍历中的位置
ansMap = new HashMap<>();
for (int i = 0; i < n; i++){
ansMap.put(inorder[i], i);
}
return createTree(preorder, inorder,0, n-1, 0, n-1);
}
//分治的思想
private TreeNode createTree(int[] preorder, int[] inorder, int preorder_left, int preorder_right, int inorder_left, int inorder_right) {
if (preorder_left > preorder_right){
return null;
}
//前序遍历中的第一个节点就是当前树的根节点
int preorder_root = preorder_left;
//定位中序遍历中的根节点,从而对中序遍历得到的数组进行切分
int inorder_root = ansMap.get(preorder[preorder_left]);
//将其造为根节点对象
TreeNode root = new TreeNode(preorder[preorder_left]);
//得到左子树的节点个数
int left_length = inorder_root - inorder_left;
//递归去构造左子树,并将其连接到根节点
//注意此处之前顺序遍历过的前序遍历数组中的元素就不要在进行遍历了,因为作为根节点的它已经被遍历过了
//前序遍历「 从左边界+1 开始到 左边界+左子树节点数目」的元素就对应了中序遍历中「从 左边界 到 根节点-1」的元素
root.left = createTree(preorder, inorder,preorder_left+1,preorder_left+left_length,inorder_left,inorder_root-1);
//递归去构造右子树,并将其连接到根节点
root.right = createTree(preorder,inorder,preorder_left+left_length+1,preorder_right,inorder_root+1,inorder_right);
return root;
}
}