[leetcode] 105. Construct Binary Tree from Preorder and Inorder Traversal (Medium)

本文介绍了一种通过先序和中序遍历构建二叉树的方法。利用递归思路,从先序遍历中选取根节点,并在中序遍历中确定其位置,以此划分左右子树,再对左右子树进行相同操作,直至构建完整棵树。

原题

题意:
根据先序和中序得到二叉树(假设无重复数字)

思路:
先手写一次转换过程,得到思路。
即从先序中遍历每个元素,(创建一个全局索引,指向当前遍历到的元素)在中序中找到该元素作为当前的root,以该节点左边所有元素作为当前root的左支,右同理。
重复分别对左右边所有元素做相同处理。

class Solution
{
public:
  TreeNode *buildTree(vector<int> &preorder, vector<int> &inorder)
  {
    int pos = 0;
    return buildBinary(preorder, inorder, 0, preorder.size(), pos);
  }

private:
  TreeNode *buildBinary(vector<int> &preorder, vector<int> &inorder, int beg,
                        int end, int &pos)
  {
    TreeNode *node = NULL;
    if (beg < end)
    {
      int i = 0;
      for (i = beg; i < end; i++)
      {
        if (preorder[pos] == inorder[i])
          break;
      }

      ++pos;
      node = new TreeNode(inorder[i]);
      node->left = buildBinary(preorder, inorder, beg, i, pos);
      node->right = buildBinary(preorder, inorder, i + 1, end, pos);
    }
    return node;
  }
};
下面是该段代码的 **Visual Studio 可运行完整版本**,包含: - `TreeNode` 节点定义 - `buildTree` 函数的完整实现(从前序和中序遍历构造二叉树) - 主函数中测试用例 - 打印构建后的二叉树(前序遍历验证) --- ## ✅ Visual Studio 完整运行代码 ```cpp #include <iostream> #include <vector> #include <algorithm> #include <ranges> using namespace std; // 二叉树节点定义 struct TreeNode { int val; TreeNode* left; TreeNode* right; TreeNode(int x) : val(x), left(nullptr), right(nullptr) {} TreeNode(int x, TreeNode* left, TreeNode* right) : val(x), left(left), right(right) {} }; class Solution { public: TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) { if (preorder.empty()) { return nullptr; } // 查找根节点在中序遍历中的位置 auto root_it = ranges::find(inorder, preorder[0]); int left_size = distance(inorder.begin(), root_it); // 左子树节点数量 // 划分左子树和右子树的前序和中序序列 vector<int> pre_left(preorder.begin() + 1, preorder.begin() + 1 + left_size); vector<int> pre_right(preorder.begin() + 1 + left_size, preorder.end()); vector<int> in_left(inorder.begin(), inorder.begin() + left_size); vector<int> in_right(inorder.begin() + 1 + left_size, inorder.end()); // 递归构建左右子树 TreeNode* left = buildTree(pre_left, in_left); TreeNode* right = buildTree(pre_right, in_right); return new TreeNode(preorder[0], left, right); } // 前序遍历打印二叉树 void preorderPrint(TreeNode* root) { if (root == nullptr) { cout << "null "; return; } cout << root->val << " "; preorderPrint(root->left); preorderPrint(root->right); } // 析构函数(手动释放内存) void deleteTree(TreeNode* root) { if (root == nullptr) return; deleteTree(root->left); deleteTree(root->right); delete root; } }; int main() { Solution sol; // 示例输入 vector<int> preorder = {3, 9, 20, 15, 7}; vector<int> inorder = {9, 3, 15, 20, 7}; // 构建二叉树 TreeNode* root = sol.buildTree(preorder, inorder); // 打印前序遍历结果以验证是否正确 cout << "构建的二叉树前序遍历结果: "; sol.preorderPrint(root); cout << endl; // 释放内存 sol.deleteTree(root); return 0; } ``` --- ## 📌 编译与运行说明(适用于 Visual Studio) 1. 打开 Visual Studio 2. 创建一个 **C++ 控制台应用程序(Console Application)** 3. 删除默认生成的代码内容 4. 将上面的代码粘贴到 `.cpp` 文件中(如 `main.cpp`) 5. 确保你的编译器支持 **C++20**,因为使用了 `<ranges>` 和范围查找 - VS 2022 及以上版本支持 C++20 6. 按 `Ctrl + F5` 或点击 **“本地 Windows 调试器”** 运行程序 --- ## 🧾 输出结果示例 ``` 构建的二叉树前序遍历结果: 3 9 null null 20 15 null null 7 null null ``` 说明构建成功,结构如下: ``` 3 / \ 9 20 / \ 15 7 ``` --- ## 🧠 代码解释 - `preorder[0]` 是当前子树的根节点。 - 在 `inorder` 中找到该根节点,左边就是左子树,右边就是右子树。 - 根据左子树大小,划分 `preorder` 中的左右子树。 - 递归构造左右子树。 - 最后返回构造好的当前子树根节点。 --- ###
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值