从零开始写算法 —— 二叉树篇 1:二叉树的三种遍历(递归实现法)

最经典的三种遍历方式是:前序中序后序

其实这三种的区别在于:"根节点(Root)"是在什么时候被访问的。

核心口诀:

  • 序: -> 左 -> 右

  • 序:左 -> -> 右

  • 序:左 -> 右 ->

  • (注:左孩子永远在右孩子前面)


前置. 递归代码的骨架

递归是解决二叉树问题最自然的方式,因为二叉树本身的定义就是递归的。

我们使用一个成员变量 ans 来存储结果,再写一个辅助函数 dfs 来进行递归。这种写法代码结构最清晰,不需要在递归函数里疯狂传递参数。

代码骨架如下:

C++代码实现:

cpp 复制代码
class Solution {
public:
    vector<int> ans; // 1. 定义结果集(放在外面,所有函数都能用)

    // 2. 递归逻辑函数 (DFS)
    void dfs(TreeNode* root) {
        // 递归终止条件:遇到空节点,直接返回
        if (root == nullptr) return;

        // --- 代码位置 A (前序) ---
        
        dfs(root->left);  // 递归处理左子树
        
        // --- 代码位置 B (中序) ---
        
        dfs(root->right); // 递归处理右子树
        
        // --- 代码位置 C (后序) ---
    }

    // 3. 主函数
    vector<int> X_Traversal(TreeNode* root) {
        dfs(root);
        return ans;
    }
};

1. 前序遍历 (Preorder Traversal)

  • 顺序 -> 左 -> 右

  • 逻辑:一进门先记录当前节点的值,然后再去管左右孩子。

  • 应用场景:复制二叉树、打印目录结构(先打印文件夹名,再打印里面的文件)。

代码实现:

C++代码实现:

cpp 复制代码
class Solution {
public:
    vector<int> ans;

    void dfs(TreeNode* root) {
        if (root == nullptr) return;

        // 【前序位置】:先记录根节点
        ans.push_back(root->val); 
        
        dfs(root->left);  // 再去左边
        dfs(root->right); // 最后去右边
    }

    vector<int> preorderTraversal(TreeNode* root) {
        dfs(root);
        return ans;
    }
};

2. 中序遍历 (Inorder Traversal)

  • 顺序 :左 -> -> 右

  • 逻辑:先一直往左走到底,回头的时候记录当前节点,最后去右边。

  • 应用场景二叉搜索树 (BST) 。BST的中序遍历结果是一个有序数组(从小到大排列),这是面试中的超高频考点!

代码实现(基于你的示例):

C++代码实现:

cpp 复制代码
class Solution {
public:
    vector<int> ans;

    void dfs(TreeNode* root) {
        if (root == nullptr) return;

        dfs(root->left);  // 先把左边走完
        
        // 【中序位置】:左边回来了,记录中间节点
        ans.push_back(root->val); 
        
        dfs(root->right); // 再去右边
    }

    vector<int> inorderTraversal(TreeNode* root) {
        dfs(root);
        return ans;
    }
};

3. 后序遍历 (Postorder Traversal)

  • 顺序 :左 -> 右 ->

  • 逻辑:先把左右孩子都处理完了,最后才处理自己。

  • 应用场景删除二叉树(必须先删掉左右孩子,才能删掉父节点,否则内存泄漏)、计算子树的大小/高度。

代码实现:

C++代码实现:

cpp 复制代码
class Solution {
public:
    vector<int> ans;

    void dfs(TreeNode* root) {
        if (root == nullptr) return;

        dfs(root->left);  // 先把左边走完
        dfs(root->right); // 把右边也走完
        
        // 【后序位置】:左右都搞定了,最后记录根节点
        ans.push_back(root->val); 
    }

    vector<int> postorderTraversal(TreeNode* root) {
        dfs(root);
        return ans;
    }
};

总结

你看,这三种看似复杂的遍历,在递归代码中仅仅是 ans.push_back(root->val) 这一行代码放的位置不同

  • 放在最前面 = 前序 (Pre)

  • 夹在中间 = 中序 (In)

  • 放在最后面 = 后序 (Post)

相关推荐
子夜江寒2 小时前
逻辑森林与贝叶斯算法简介
算法·机器学习
xu_yule3 小时前
算法基础-背包问题(01背包问题)
数据结构·c++·算法·01背包
我不是小upper3 小时前
从理论到代码:随机森林 + GBDT+LightGBM 融合建模解决回归问题
人工智能·深度学习·算法·随机森林·机器学习·回归
budingxiaomoli3 小时前
分治算法-快排
数据结构·算法
云泽8083 小时前
C++ STL 栈与队列完全指南:从容器使用到算法实现
开发语言·c++·算法
前端小白在前进4 小时前
力扣刷题:复原IP地址
tcp/ip·算法·leetcode
yaoh.wang4 小时前
力扣(LeetCode) 94: 二叉树的中序遍历 - 解法思路
python·算法·leetcode·面试·职场和发展·二叉树·跳槽
Evand J4 小时前
【课题推荐】基于视觉(像素坐标)与 IMU 的目标/自身运动估计(Visual-Inertial Odometry, VIO),课题介绍与算法示例
人工智能·算法·计算机视觉
xu_yule4 小时前
算法基础(背包问题)—分组背包和混合背包
c++·算法·动态规划·分组背包·混合背包