根据一棵树的前序遍历与中序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。
例如,给出
前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
3
/ \
9 20
/ \
15 7
通过前序遍历可以看出,根节点都是数组中的第一个,从中序遍历中可以看出,左子树都在根节点的左边,右子树都在根节点的右边。所以可以递归。通过HashMap来存储中序遍历,key为元素的值,value为下标,这样我们通过根节点的值就可以快速的访问到根节点在中序遍历中的位置,从而划分左右子树。
递归方法的前两个参数是在前序遍历中,子树的开始位置和结束位置,后两个参数是在中序遍历中,子树的开始位置和结束位置
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
HashMap<Integer,Integer> map = new HashMap<>();
int[] pre;
public TreeNode buildTree(int[] preorder, int[] inorder) {
for (int i = 0; i < inorder.length; i++)
{
map.put(inorder[i],i);
}
pre = preorder;
TreeNode tree = createTree(0,pre.length-1,0,inorder.length-1);
return tree;
}
public TreeNode createTree(int ps, int pe, int is, int ie)
{
if (ps > pe || is > ie)
return null;
int ri = pre[ps];
// 头结点在inorder数组中的位置
int root = map.get(ri);
TreeNode node = new TreeNode(ri);
// 递归得到左子树
node.left = createTree(ps+1,ps+root-is,is,root);
// 递归得到右子树
node.right = createTree(ps+root-is+1,pe,root+1,ie);
return node;
}
}