从前序与中序遍历序列构造二叉树(C++实现)

简介: 从前序与中序遍历序列构造二叉树(C++实现)


题目

思路分析

  • 我们可以通过递归实现的二叉树构建函数。它根据给定的先序遍历序列和中序遍历序列构建一棵二叉树,并返回根节点
  • 可以创建一个_build 函数,该函数负责构建二叉树的节点,通过分割先序遍历序列和中序遍历序列,并递归构建左子树和右子树来完成整个二叉树的构建过程。
  • 最终,buildTree 函数调用 _build 函数,并返回构建的二叉树的根节点。

代码

class Solution {
public:
    TreeNode* _build(vector<int>& preorder, vector<int>& inorder, int& peri, int begin, int end) 
    {
         // 如果开始索引大于结束索引,则返回空指针,表示当前子树为空
         if (begin > end) {
             return nullptr;
         }
         int rooti = 0;
         // 在中序遍历序列中找到当前根节点的索引
         while (rooti <= end) {
             if (inorder[rooti] == preorder[peri]) {
                 break;
             }
             rooti++;
         }
         // 创建当前根节点
         TreeNode* root = new TreeNode(preorder[peri++]);
         // 构建左子树
         // 左子树的先序遍历序列为 [begin, rooti-1]
         // 左子树的中序遍历序列为 [begin, rooti-1]
         root->left = _build(preorder, inorder, peri, begin, rooti-1);
         // 构建右子树
         // 右子树的先序遍历序列为 [rooti+1, end]
         // 右子树的中序遍历序列为 [rooti+1, end]
         root->right = _build(preorder, inorder, peri, rooti+1, end);
         return root;
    }
    TreeNode* buildTree(vector<int>& preorder, vector<int>& inorder) 
    {
        int i = 0;
        // 调用内部函数 _build 构建二叉树
        // 初始时 peri 为 0,表示从先序遍历序列的第一个元素开始构建
        // 根据先序遍历序列和中序遍历序列构建二叉树,并返回根节点
        TreeNode* root = _build(preorder, inorder, i, 0, preorder.size()-1);
        return root;
    }
};

代码讲解

首先判断起始索引 begin 是否大于结束索引 end,如果是,则说明当前子树为空,返回空指针。

在中序遍历序列 inorder 中找到当前根节点的索引 rooti,该索引表示根节点在中序遍历中的位置。在循环中,不断遍历中序遍历序列,直到找到与先序遍历序列 preorder[peri] 相等的值。找到后,rooti 就是根节点的索引。

创建当前根节点,节点的值为 preorder[peri]。

根据先序遍历序列和中序遍历序列的特性,可以将当前子树分为左子树和右子树。左子树的先序遍历序列为 preorder[begin, rooti-1],左子树的中序遍历序列为 inorder[begin, rooti-1]。右子树的先序遍历序列为 preorder[rooti+1, end],右子树的中序遍历序列为 inorder[rooti+1, end]。

递归调用 _build 函数,分别构建左子树和右子树,并将返回的根节点设置为当前根节点的左孩子和右孩子。

返回当前根节点。


总结:

在 buildTree 函数中,首先定义了一个变量 i,初始值为 0,作为 _build 函数中 peri 的初始值。然后调用 _build 函数,传入先序遍历序列 preorder、中序遍历序列 inorder、i、0 和 preorder.size()-1,即整个序列的起始和结束索引。最后返回构建的二叉树的根节点。

相关文章
|
3月前
|
传感器 定位技术 C++
基于C++的GDAL用空白栅格填充长时间序列遥感影像中的缺失图像
然后,定义需要处理的遥感影像路径列表,和识别数据缺失的逻辑。这里我们简化处理,假设已经知道哪一幅图像是缺失的,因此直接跳过识别步骤。
57 1
|
3月前
|
JavaScript 前端开发 Java
面向对象编程(C++篇2)——构造
面向对象编程(C++篇2)——构造
31 0
|
4月前
|
存储 C++
【C++】二叉树进阶之二叉搜索树(下)
【C++】二叉树进阶之二叉搜索树(下)
32 4
|
4月前
|
Java 编译器 C++
【C++】二叉树进阶之二叉搜索树(上)
【C++】二叉树进阶之二叉搜索树(上)
36 3
|
4月前
|
算法 C++
【C++高阶】高效搜索的秘密:深入解析搜索二叉树
【C++高阶】高效搜索的秘密:深入解析搜索二叉树
42 2
|
4月前
|
存储 算法 搜索推荐
|
5月前
|
C++ 容器
C++之deque容器(构造、赋值、大小、插入与删除、存取、排序)
C++之deque容器(构造、赋值、大小、插入与删除、存取、排序)
|
5月前
|
算法 搜索推荐 C++
C++之STL常用算法(遍历、查找、排序、拷贝、替换、算数生成、集合)
C++之STL常用算法(遍历、查找、排序、拷贝、替换、算数生成、集合)
|
7天前
|
存储 编译器 C++
【c++】类和对象(中)(构造函数、析构函数、拷贝构造、赋值重载)
本文深入探讨了C++类的默认成员函数,包括构造函数、析构函数、拷贝构造函数和赋值重载。构造函数用于对象的初始化,析构函数用于对象销毁时的资源清理,拷贝构造函数用于对象的拷贝,赋值重载用于已存在对象的赋值。文章详细介绍了每个函数的特点、使用方法及注意事项,并提供了代码示例。这些默认成员函数确保了资源的正确管理和对象状态的维护。
34 4
|
8天前
|
存储 编译器 Linux
【c++】类和对象(上)(类的定义格式、访问限定符、类域、类的实例化、对象的内存大小、this指针)
本文介绍了C++中的类和对象,包括类的概念、定义格式、访问限定符、类域、对象的创建及内存大小、以及this指针。通过示例代码详细解释了类的定义、成员函数和成员变量的作用,以及如何使用访问限定符控制成员的访问权限。此外,还讨论了对象的内存分配规则和this指针的使用场景,帮助读者深入理解面向对象编程的核心概念。
31 4