【Hot 100】94. 二叉树的中序遍历

目录

  • 🙋‍♂️ 作者:海码007
  • 📜 专栏:算法专栏
  • 💥 标题:【Hot 100】94. 二叉树的中序遍历
  • ❣️ 寄语:书到用时方恨少,事非经过不知难!

引言

今天开始二叉树的篇章,继续加油。

二叉树的中序遍历

我的解题

二叉树的遍历有四种,分别是前序、中序、后序以及层次遍历。前中后序遍历可以通过递归写出清晰的代码,当然也可以通过栈来写出非递归的代码。然后是层次遍历通过借助队列来实现一层一层的遍历顺序。

下面来解析一下我的代码,本题给出的核心函数是 inorderTraversal 然后函数有一个返回值,这个返回值记录的是二叉树的值。如果我直接将 inorderTraversal 作为一个递归函数来写的话,就需要处理这个返回值,要考虑如何将这个返回值合并。所以为了让记录变得更加方便我新建了一个递归函数 midTraversal ,这个函数没有返回值,但是使用了引用传递来让这个结果合并起来。

cpp 复制代码
class Solution {
public:
    void midTraversal(TreeNode* node, vector<int>& data)
    {
        if (!node) return;
        midTraversal(node->left, data);
        data.push_back(node->val);
        midTraversal(node->right, data);
    }

    vector<int> inorderTraversal(TreeNode* root) {
        vector<int> res;
        midTraversal(root, res);
        return res;
    }
};

代码优化

  1. 递归函数的返回值问题

    • 如果直接让 inorderTraversal 递归调用自身,每次递归都需要合并子树的遍历结果(即处理返回值),这会增加代码复杂度。

    • 例如:

      cpp 复制代码
      vector<int> inorderTraversal(TreeNode* root) {
          if (!root) return {};
          auto left = inorderTraversal(root->left);  // 需要合并左子树的结果
          auto right = inorderTraversal(root->right); // 需要合并右子树的结果
          left.push_back(root->val);                // 插入当前节点
          left.insert(left.end(), right.begin(), right.end()); // 合并右子树
          return left; // 返回合并后的结果
      }
    • 这种写法需要频繁合并向量,效率较低(时间复杂度为 O(n^2)),且代码冗余。

  2. 引用传递的优化

    • 通过引入辅助函数 midTraversal,将结果向量 data 通过引用传递,避免返回值合并的问题。
    • 递归过程中直接修改同一个 data 向量,无需合并,效率更高(时间复杂度 O(n))。

更清晰的表述建议:

"中序遍历的递归实现需要按左子树->根节点->右子树的顺序访问节点。如果直接在 inorderTraversal 中递归,每次递归调用会返回一个独立的向量,需要将这些向量合并(如拼接左子树、当前节点、右子树的结果),效率较低。

因此,我引入辅助函数 midTraversal,通过引用传递一个共享的结果向量 data。递归过程中,所有节点值直接追加到 data 中,无需合并操作,既简化了代码,又保证了线性时间复杂度。"

相关推荐
PAK向日葵2 小时前
【算法导论】PDD 0817笔试题题解
算法·面试
地平线开发者4 小时前
ReID/OSNet 算法模型量化转换实践
算法·自动驾驶
快乐的划水a5 小时前
组合模式及优化
c++·设计模式·组合模式
地平线开发者5 小时前
开发者说|EmbodiedGen:为具身智能打造可交互3D世界生成引擎
算法·自动驾驶
星星火柴9366 小时前
关于“双指针法“的总结
数据结构·c++·笔记·学习·算法
艾莉丝努力练剑7 小时前
【洛谷刷题】用C语言和C++做一些入门题,练习洛谷IDE模式:分支机构(一)
c语言·开发语言·数据结构·c++·学习·算法
闪电麦坤958 小时前
数据结构:迭代方法(Iteration)实现树的遍历
数据结构·二叉树·
C++、Java和Python的菜鸟8 小时前
第六章 统计初步
算法·机器学习·概率论
Cx330❀8 小时前
【数据结构初阶】--排序(五):计数排序,排序算法复杂度对比和稳定性分析
c语言·数据结构·经验分享·笔记·算法·排序算法
散1128 小时前
01数据结构-Prim算法
数据结构·算法·图论