目录
一、题目描述
二、整体思路
三、代码
一、题目描述
题目地址
二、整体思路
前序遍历得到的是[根结点|左子树|右子树],中序遍历得到的是[左子树|根结点|右子树]
那么可以设立一个递归函数,作用是利用前序遍历的数组和中序遍历的数组构建一个节点的二叉树
参数为(前序遍历数组、中序遍历数组、前序遍历数组的左边界、前序遍历数组的右边界、中序遍历数组的左边界、中序遍历数组的右边界)
问题的关键是要找到根节点,每一次调用递归函数时,前序遍历数组的左边界为根节点的索引,然后我们要设法找到根节点在中序遍历数组的索引。因此,一开始可以先用哈希表保存根节点在中序遍历数组的索引。
根节点在Preorder中索引永远为l1,在Inorder索引i可以通过哈希表得到
Inorder中[l2,i-1]的元素就是左子树的元素,在Preorder中表现为[l1+1,l1+i-l2]
同理,Inorder中[i+1,r2]的元素就是右子树的元素,在Preorder中表现为[l1+i-l2+1,r1](l1+i-l2为左子树元素,+1就是右子树第一个元素)
综上,Preorder中根节点索引为l1,左子树元素区间为[l1+1,l1+i-l2],右子树区间为[l1+i-l2+1,r1]
Inorder中根节点索引为i,左子树元素区间为[l2,i-1],右子树区间为[i+1,r2]
三、代码
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
Map<Integer,Integer> map=new HashMap<>();
public TreeNode buildTree(int[] preorder, int[] inorder) {
if(preorder.length==1 && inorder.length==1){
return new TreeNode(preorder[0]);
}
for(int i=0;i<inorder.length;i++){
map.put(inorder[i],i);
}
TreeNode root=build(preorder,inorder,0,preorder.length-1,0,inorder.length-1);
return root;
}
public TreeNode build(int[] preorder,int[] inorder,int l1,int r1,int l2,int r2){
if(l1>r1 || l2>r2){
return null;
}
TreeNode node=new TreeNode(preorder[l1]);
int newl1=map.get(preorder[l1]);
node.left=build(preorder,inorder,l1+1,l1+newl1-l2,l2,newl1-1);
node.right=build(preorder,inorder,l1+newl1-l2+1,r1,newl1+1,r2);
return node;
}
}