思路:使用递归实现,想明白这里还比较容易,个人觉得难点在于如何设计这个递归方法,它的参数应该有哪些:也就是应该用哪些参数来界定子问题。
该算法的关键是
public TreeNode build(int[] preorder, int preStart, int preEnd, int[] inorder, int inStart, int inEnd)
该递归方法有六个参数:前序遍历和中序遍历数组,以及每个子树在这两个数组中对应的起始位置。
难点二在于:左右子树的参数值应该是什么,这几个参数值,结合网上的博客加上自己的思考也不是很难理解。
Java代码:
public class Solution {
/**
* 使用前序遍历和中序遍历数组递归构造二叉树的递归方法实现
* @param preorder 前序遍历数组
* @param preStart preStart用于确定这个子树的前序遍历序列在原始前序遍历数组中的起始位置
* @param preEnd preEnd用于确定这个子树的前序遍历序列在原始前序遍历数组中的结束位置
* @param inorder 中序遍历数组
* @param inStart inStart用于确定这个子树的中序遍历序列在原始中序遍历数组中的起始位置
* @param inEnd inEnd用于确定这个子树的中序遍历序列在原始中序遍历数组中的结束位置
* @return
*/
public TreeNode build(int[] preorder, int preStart, int preEnd, int[] inorder, int inStart, int inEnd) {
// 递归结束的条件(越界 or 子树长度小于1)
if (preStart >= preorder.length || preEnd < preStart) {
return null;
}
TreeNode root = new TreeNode(preorder[preStart]);
int index = 0;
// 获取中序遍历中根节点的下标
for (int i = inStart; i <= inEnd; i++) {
if (inorder[i] == preorder[preStart]) {
index = i;
break;
}
}
// 子树节点长度
int len = index - inStart;
// 重点是preStart、preEnd、inStart、 inEnd四个参数的取值
root.left = build(preorder, preStart + 1, preStart + len, inorder, inStart, index - 1);
root.right = build(preorder, preStart + len + 1, preEnd, inorder, index + 1, inEnd);
return root;
}
public TreeNode buildTree(int[] preorder, int[] inorder) {
if (preorder == null || inorder == null) {
return null;
}
if (preorder.length == 0 || inorder.length == 0) {
return null;
}
return build(preorder, 0, preorder.length - 1, inorder, 0, inorder.length - 1);
}
}