【Hot 100 刷题计划】 LeetCode 102. 二叉树的层序遍历 | C++ 极简 DFS 巧解

LeetCode 102. 二叉树的层序遍历

📌 题目描述

题目级别:中等

给你二叉树的根节点 root ,返回其节点值的 层序遍历 。 (即逐层地,从左到右访问所有节点)。

  • 示例 1:
    输入:root = [3,9,20,null,null,15,7]
    输出:[[3],[9,20],[15,7]]

💡 破题思路:DFS 携带深度参数

虽然题目是"层序遍历",但我们完全可以用深度优先搜索(DFS)来"降维打击"。

核心思路是:在 DFS 向下递归的时候,顺便带上一个表示当前深度的参数 level

当我们走到某一个节点时:

  1. 如果发现结果集 res 的大小正好等于 level,说明当前这一层还没有专属的数组,我们立刻在 res 中压入一个空数组。
  2. 然后,把当前节点的值,精准地放入 res[level] 中。

⚠️ 面试避坑点:

不要将 res 设为类的成员变量(全局变量),否则在处理多个测试用例时,上一次残留的数据会导致结果全部错乱。正确的做法是在主函数中声明,并通过引用传递给 DFS 函数。


💻 C++ 代码实现 (DFS 降维打击版)

cpp 复制代码
class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        vector<vector<int>> res; // 局部变量,保证每次调用都是全新的状态
        dfs(root, 0, res);
        return res;
    }

    // 将 res 以引用形式传入
    void dfs(TreeNode* root, int level, vector<vector<int>>& res)
    {
        if (!root) return;

        // 如果这是该层第一次被访问到,为该层开辟一个新的一维数组
        if (res.size() == level) {
            res.push_back(vector<int>());
        }

        // 将当前节点按层号对号入座
        res[level].push_back(root->val);

        // 递归遍历左右子树,层号 +1
        dfs(root->left, level + 1, res);
        dfs(root->right, level + 1, res);
    }
};
相关推荐
赴生-34 分钟前
C++进阶 C++11(下)
开发语言·c++
有点。38 分钟前
C++(贪心算法一)
c++·贪心算法
WBluuue1 小时前
数据结构与算法:有序表(二):跳表
数据结构·c++·算法·skiplist
赴生-1 小时前
C++进阶 异常
开发语言·c++
凡人叶枫2 小时前
Effective C++ 条款28:避免使用 handles 指向对象内部
linux·服务器·开发语言·c++·嵌入式开发
zwenqiyu3 小时前
P5283 [十二省联考 2019] 异或粽子题解
c++·学习·算法
Queenie_Charlie3 小时前
哈夫曼树
数据结构·c++·哈夫曼树
lihao lihao3 小时前
Linux信号
开发语言·c++·算法
大白话_NOI4 小时前
【洛谷 P2249】查找(深基 13. 例 1)+ 详细分析
c++·算法
智者知已应修善业4 小时前
【51单片机2个外部中断显示中断历时,初始化8左移3位共阳数码管】2024-6-6
c++·经验分享·笔记·算法·51单片机