从零开始写算法 —— 二叉树篇 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)

相关推荐
醉颜凉几秒前
【LeetCode】打家劫舍III
c语言·算法·leetcode·树 深度优先搜索·动态规划 二叉树
达文汐3 分钟前
【困难】力扣算法题解析LeetCode332:重新安排行程
java·数据结构·经验分享·算法·leetcode·力扣
一匹电信狗3 分钟前
【LeetCode_21】合并两个有序链表
c语言·开发语言·数据结构·c++·算法·leetcode·stl
User_芊芊君子3 分钟前
【LeetCode经典题解】搞定二叉树最近公共祖先:递归法+栈存路径法,附代码实现
算法·leetcode·职场和发展
算法_小学生5 分钟前
LeetCode 热题 100(分享最简单易懂的Python代码!)
python·算法·leetcode
执着2595 分钟前
力扣hot100 - 234、回文链表
算法·leetcode·链表
Gorgous—l7 分钟前
数据结构算法学习:LeetCode热题100-多维动态规划篇(不同路径、最小路径和、最长回文子串、最长公共子序列、编辑距离)
数据结构·学习·算法
熬夜造bug7 分钟前
LeetCode Hot100 刷题路线(Python版)
算法·leetcode·职场和发展
2401_8384725129 分钟前
C++中的访问者模式
开发语言·c++·算法
老鼠只爱大米35 分钟前
LeetCode经典算法面试题 #108:将有序数组转换为二叉搜索树(递归分治、迭代法等多种实现方案详解)
算法·leetcode·二叉树·二叉搜索树·平衡树·分治法