剑指 offer 树算法题:重建二叉树

给定二叉树的前序和中序遍历序列,可以使用递归方法重建二叉树。首先找到前序遍历中的第一个节点,它是中序遍历中的分隔符,然后分别构造左右子树。同样,后序遍历结合中序遍历也可实现重建,但仅凭前序和后序遍历无法唯一确定二叉树。
摘要由CSDN通过智能技术生成

题目描述:输入某二叉树的前序遍历和中序遍历的结果,请构建该二叉树并返回其根节点。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列[1,2,4,7,3,5,6,8]和中序遍历序列[4,7,2,1,5,3,8,6],则重建二叉树并返回。

分析:在中序遍历序列中找到前序遍历序列的第一个节点1,该节点将中序遍历序列分成左子树(中序遍历序列)[4,7,2]右子树(中序遍历序列)[5,3,8,6]两个部分,而在前序遍历序列中紧接着的和左子树相同长度的序列即为左子树的前序遍历序列[2,4,7]以及和右子树相同长度的序列即为右子树的前序遍历序列[3,5,6,8],如此递归下去。

如果给出的是后序和中序,同理,只不过后序的根节点在最后。

但是只给出前序和后序是无法确定一颗二叉树的。

求解

// 前序 + 中序
function buildTree1(preorder: number[], inorder: number[]): TreeNode<number> | null {
  const size = preorder.length;
  if (size === 0) return null;
  // 前序遍历的首个节点将中序遍历该节点前后划分为左子树和右子树
  // 前序遍历中首个节点紧接着的和左子树相同长度的序列即为左子树的前序遍历序列,余下的为右子树
  const rootVal = preorder[0];
  const divider = inorder.indexOf(rootVal);
  const leftPreorder = preorder.slice(1, divider + 1);
  const rightPreorder = preorder.slice(divider + 1);
  const leftInorder = inorder.slice(0, divider);
  const rightInorder = inorder.slice(divider + 1);
  const leftTree = buildTree1(leftPreorder, leftInorder);
  const rightTree = buildTree1(rightPreorder, rightInorder);
  return new TreeNode(rootVal, leftTree, rightTree);
}

// 后序 + 中序
function buildTree2(postorder: number[], inorder: number[]): TreeNode<number> | null {
  const size = postorder.length;
  if (size === 0) return null;
  // 后序遍历的最后节点将中序遍历该节点前后划分为左子树和右子树
  // 前序遍历中和左子树相同长度的序列即为左子树的前序遍历序列,余下的出最后一个节点之外的即为右子树
  const rootVal = postorder[size - 1];
  const divider = inorder.indexOf(rootVal);
  const leftPostorder = postorder.slice(0, divider);
  const rightPostorder = postorder.slice(divider, -1);
  const leftInorder = inorder.slice(0, divider);
  const rightInorder = inorder.slice(divider + 1);
  const leftTree = buildTree2(leftPostorder, leftInorder);
  const rightTree = buildTree2(rightPostorder, rightInorder);
  return new TreeNode(rootVal, leftTree, rightTree);
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

薛定谔的猫96

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值