106.从中序与后序遍历序列构造二叉树
/**
* 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 TreeNode buildTree(int[] inorder, int[] postorder) {
int n = postorder.length;
if (n == 0) { // 空节点
return null;
}
int leftSize = indexOf(inorder, postorder[n - 1]); // 左子树的大小
int[] in1 = Arrays.copyOfRange(inorder, 0, leftSize);
int[] in2 = Arrays.copyOfRange(inorder, leftSize + 1, n);
int[] post1 = Arrays.copyOfRange(postorder, 0, leftSize);
int[] post2 = Arrays.copyOfRange(postorder, leftSize, n - 1);
TreeNode left = buildTree(in1, post1);
TreeNode right = buildTree(in2, post2);
return new TreeNode(postorder[n - 1], left, right);
}
// 返回 x 在 a 中的下标,保证 x 一定在 a 中
private int indexOf(int[] a, int x) {
for (int i = 0;; i++) {
if (a[i] == x) {
return i;
}
}
}
}
105.从前序与中序遍历序列构造二叉树
- 首先,代码获取preorder数组的长度n,如果n为0,则表示当前节点为空节点,直接返回null。
- 接下来,代码通过调用indexOf函数找到preorder在inorder数组中的下标leftSize,即左子树的大小。
- 然后,代码使用Arrays.copyOfRange函数将preorder和inorder数组分成四个部分:pre1、pre2、in1、in2。其中pre1是preorder数组中除去根节点后左子树的部分,pre2是preorder数组中除去根节点后右子树的部分;in1是inorder数组中左子树的部分,in2是inorder数组中右子树的部分。
- 接着,代码递归调用buildTree函数,传入pre1和in1作为参数,构建左子树,并将返回的左子树赋值给变量left。然后,代码再次递归调用buildTree函数,传入pre2和in2作为参数,构建右子树,并将返回的右子树赋值给变量right。
- 最后,代码使用preorder作为值,left作为左子树,right作为右子树,创建一个新的TreeNode对象,并将其作为函数的返回值。
/**
* 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 TreeNode buildTree(int[] preorder, int[] inorder) {
int n = preorder.length;
if (n == 0) { // 空节点
return null;
}
int leftSize = indexOf(inorder,preorder[0]);
int[] pre1 = Arrays.copyOfRange(preorder, 1, 1 + leftSize);
int[] pre2 = Arrays.copyOfRange(preorder, 1 + leftSize, n);
int[] in1 = Arrays.copyOfRange(inorder, 0, leftSize);
int[] in2 = Arrays.copyOfRange(inorder, 1 + leftSize, n);
TreeNode left = buildTree(pre1, in1);
TreeNode right = buildTree(pre2, in2);
return new TreeNode(preorder[0], left, right);
}
// 返回 x 在 a 中的下标,保证 x 一定在 a 中
private int indexOf(int[] a, int x) {
for (int i = 0; ; i++) {
if (a[i] == x) {
return i;
}
}
}
}