一、题目
力扣原题《从前序与中序遍历序列构造二叉树》:https://leetcode-cn.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal/
力扣原题《从中序与后序遍历序列构造二叉树》:https://leetcode-cn.com/problems/construct-binary-tree-from-inorder-and-postorder-traversal/
二、递归
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public TreeNode buildTree(int[] preorder, int[] inorder) {
if (null == preorder || 0 == preorder.length || null == inorder || 0 == inorder.length) {
return null;
}
TreeNode root = new TreeNode(preorder[0]);
for (int i = 0; i < inorder.length; i++) {
if (preorder[0] != inorder[i]) {
continue;
}
root.left = buildTree(Arrays.copyOfRange(preorder, 1, i + 1),
Arrays.copyOfRange(inorder, 0, i));
root.right = buildTree(Arrays.copyOfRange(preorder, i + 1, preorder.length),
Arrays.copyOfRange(inorder, i + 1, inorder.length));
}
return root;
}
}
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public TreeNode buildTree(int[] inorder, int[] postorder) {
if (null == inorder || 0 == inorder.length || null == postorder || 0 == postorder.length) {
return null;
}
int val = postorder[postorder.length - 1];
TreeNode root = new TreeNode(val);
for (int i = 0; i < inorder.length; i++) {
// 取出后序遍历的最后一个元素,定位其在中序遍历中的索引
if (inorder[i] != val) {
continue;
}
// 递归构建左子树
root.left = buildTree(Arrays.copyOfRange(inorder, 0, i),
Arrays.copyOfRange(postorder, 0, i));
// 递归处理右子树
if (i > postorder.length - 1) {
continue;
}
root.right = buildTree(Arrays.copyOfRange(inorder, i + 1, inorder.length),
Arrays.copyOfRange(postorder, i, postorder.length - 1));
}
return root;
}
}
- 基本思路:根据前序、中序、后序遍历的定义,可以找到递归的规律。以前序 + 中序重建二叉树为例,每次前序遍历的首个节点即递归的根节点,找到该节点在中序遍历中的位置,记索引为index。可以发现,在前序、中序序列中以index分割,前序、中序的左半部分依旧是左子树的前序、中序遍历序列,右半部分依旧是右子树的前序、中序遍历序列。p.s.:有点绕,画一下就知道了。
- 时间复杂度:O(n)。遍历所有节点,重建二叉树。
- 空间复杂度:O(n)。存储整棵二叉树的开销。
三、总结
- 已知二叉树的中序 + 前序/后序遍历序列,可以还原二叉树;
- 二叉树的问题经常可以转化为递归问题;