前序和后序构造二叉树

105.从前序与中序遍历序列构造二叉树

给定两个整数数组 preorderinorder ,其中 preorder 是二叉树的先序遍历inorder 是同一棵树的中序遍历,请构造二叉树并返回其根节点。

示例 1:

img

1
2
输入: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
输出: [3,9,20,null,null,15,7]

示例 2:

1
2
输入: preorder = [-1], inorder = [-1]
输出: [-1]

提示:

  • 1 <= preorder.length <= 3000
  • inorder.length == preorder.length
  • -3000 <= preorder[i], inorder[i] <= 3000
  • preorderinorder无重复 元素
  • inorder 均出现在 preorder
  • preorder 保证 为二叉树的前序遍历序列
  • inorder 保证 为二叉树的中序遍历序列
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
class Solution {
public TreeNode buildTree(int[] preorder, int[] inorder) {
return traversal(inorder, 0 , inorder.length-1, preorder, 0, preorder.length-1);
}

public TreeNode traversal(int[] inorder, int inleft, int inright, int[]preorder, int preleft, int preright){
if(inright<inleft) return null;

//以先序序列的第一个元素为值创建节点
int rootval = preorder[preleft];
TreeNode root = new TreeNode(rootval);

int index = inleft;//找到中序遍历分割点
while(index <= inright){
if(inorder[index] == rootval){
break;
}
index++;
}
int left_len = index - inleft;//中序左数组的长度
//中序左数组长度等于前序左数组长度
root.left = traversal(inorder, inleft, index-1, preorder, preleft+1, preleft+left_len);
root.right = traversal(inorder, index+1, inright, preorder, preleft+left_len+1, preright);

return root;
}
}

前序和后序构造二叉树
http://example.com/2023/04/09/算法/二叉树/17. 中序遍历与前序遍历构造二叉树/
作者
PALE13
发布于
2023年4月9日
许可协议