面试题6:根据前序和中序序列构建二叉树(Leetcode-106)

题目:输入二叉树的前序遍历和中序遍历序列,重构其二叉树,假设输入的数字都是不重复的。
例如前序序列是 {1,2,4,7,3,5,6,8},中序序列是{4,7,2,1,5,3,8,6}。
分析如下图所示:
这里写图片描述

算法步骤:

  1. 前序序列pre的第一个元素即根节点。
  2. 遍历中序序列mid,找到的根节点,得到左子树的大小left_size。
  3. 左子树的中序序列为mid[0,left_size ),右子树的中序序列为mid[left_size+1, mid.size)
  4. 左子树的前序序列为pre[1,left_size ),右子树的前序序列为pre[left_size+1, pre.size)

代码如下:

TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
  if (pre.size() == 0 || vin.size() == 0) {
    return nullptr;
  }
  int root_val = pre[0];
  TreeNode* root = new TreeNode(root_val);
  int left_size = 0;
  for (int i = 0; i < vin.size(); ++i) {
    if (vin[i] == root_val) {
      left_size = i;
      break;
    }
  }
  vector<int> left_vin;
  left_vin.assign(vin.begin(), vin.begin() + left_size);
  vector<int> right_vin;
  right_vin.assign(vin.begin() + left_size + 1, vin.end());
  vector<int> left_pre;
  left_pre.assign(pre.begin() + 1, pre.begin() + 1 + left_size);
  vector<int> right_pre;
  right_pre.assign(pre.begin() + 1 + left_size, pre.end());

  root->left = reConstructBinaryTree(left_pre, left_vin);
  root->right = reConstructBinaryTree(right_pre, right_vin);
  return root;
}
  • 1
    点赞
  • 15
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值