题目描述
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
解题思路
- 前序遍历的第一位是根节点。
- 中序遍历中,根节点左右两边分别是左子树和右子树。
代码
public class Solution {
public TreeNode reConstructBinaryTree(int[] pre, int[] in) {
// 边界条件
if (pre == null || in == null || pre.length != in.length) {
return null;
}
return construct(pre, 0, pre.length - 1, in, 0, in.length - 1);
}
public TreeNode construct(int[] pre, int preS, int preE, int[] in, int inS, int inE) {
if (preS > preE || inS > inE)
return null;
// 前序遍历的第一个是根节点,创建根节点
TreeNode node = new TreeNode(pre[preS]);
for (int i = inS; i <= inE; i++) {
// 在中序遍历中寻找根节点
if (in[i] == pre[preS]) {
// 递归调用,根据当前根节点构建左子树
node.left = construct(pre, preS + 1, preS + i - inS, in, inS, i - 1);
// 递归调用,根据当前根节点构建右子树
node.right = construct(pre, preS + i - inS + 1, preE, in, i + 1, inE);
}
}
return node;
}
}
文章汇总见这里 >>剑指Offer汇总<<
更多精彩文章见这里 >>我的学习小站<< & 更多好玩见这里 >>技术杂谈<<