力扣链接:105. 从前序与中序遍历序列构造二叉树 - 力扣(LeetCode)
问题主体:
给定两个整数数组 preorder
和 inorder
,其中 preorder
是二叉树的先序遍历, inorder
是同一棵树的中序遍历,请构造二叉树并返回其根节点。
示例 1:
输入: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7] 输出: [3,9,20,null,null,15,7]
示例 2:
输入: preorder = [-1], inorder = [-1] 输出: [-1]
提示:
1 <= preorder.length <= 3000
inorder.length == preorder.length
-3000 <= preorder[i], inorder[i] <= 3000
preorder
和inorder
均 无重复 元素inorder
均出现在preorder
preorder
保证 为二叉树的前序遍历序列inorder
保证 为二叉树的中序遍历序列
题解:
该问题要我们根据前序遍历和中序遍历构造出一个二叉树,
前序遍历的遍历过程是:根,左,右
中序遍历的遍历过程是: 左,根,右
因此我们可以从前序遍历的过程中获得根节点,从中序遍历的过程中将根节点的左右子树分出来
递归这个过程,最终完成二叉树的创建
代码实现:
/**
* 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 {
public int index;
public TreeNode buildTree(int[] preorder, int[] inorder) {
return buildTreeChild(preorder,inorder,0,inorder.length-1);
}
private TreeNode buildTreeChild(int[] preorder, int[] inorder,int inorderBegin,int inorderEnd) {
if(inorderBegin > inorderEnd){
return null;
}
TreeNode root = new TreeNode(preorder[index]);
int rootindex = findIndec(inorder,inorderBegin,inorderEnd,preorder[index]);
if(rootindex == -1){
return null;
}
index++;
root.left =buildTreeChild(preorder,inorder,inorderBegin,rootindex-1);
root.right = buildTreeChild(preorder,inorder,rootindex+1,inorderEnd);
return root;
}
private int findIndec(int[] inorder,int inorderBegin,int inorderEnd,int key){
for(int i = inorderBegin;i<=inorderEnd;i++){
if(inorder[i] == key){
return i;
}
}
return -1;
}
}