leecode429 | N叉树层序遍历 | dfs | 年过得真快

最新推荐文章于 2025-12-04 16:02:12 发布

原创最新推荐文章于 2025-12-04 16:02:12 发布 · 237 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#深度优先 #算法

综合部同时被 3 个专栏收录

130 篇文章

订阅专栏

C++

92 篇文章

订阅专栏

leecode

42 篇文章

订阅专栏

本文讨论了使用迭代和队列进行树的层次遍历的方法，重点介绍了如何通过记录层号降低空间复杂度，并提到了递归函数的应用。作者反思了年前错过练习的经历，感慨时间流逝和个人成长的需求。

树的层次遍历，其实如果用迭代算是简单题吧
层序遍历，常规思路就是用队列，保存每次父节点的下一层子节点，然后在遍历这一层的子节点采用队列的思想，先进后出
然后看到有的题解，采用 index 记录 “层号” 这样空间复杂度会低一些

在这里插入图片描述

/*
// Definition for a Node.
class Node {
public:
    int val;
    vector<Node*> children;

    Node() {}

    Node(int _val) {
        val = _val;
    }

    Node(int _val, vector<Node*> _children) {
        val = _val;
        children = _children;
    }
};
*/

class Solution {
public:
    vector<vector<int>> levelOrder(Node* root) {
        std::vector<std::vector<int>> res;
        if(!root){
            return res;
        }
        function<void(Node*, int)> dfs = [&](Node* cur, int lev){
            if(!cur){
                return;
            }
            if(lev == res.size()){
                res.push_back({});
            }
            res[lev].push_back(cur->val);
            for(auto i : cur->children){
                dfs(i, lev + 1);
            }
        };
		// dfs(根节点，节点所在的层数）  倒置 从0开始
        dfs(root, 0);
        return res;
    }
};

#这里在详细讲一下吧， 发现function 函数还是用的少

#在遍历每一层的子节点，该层的lev 都是一样的，只有当该子节点还有孩子节点，则会继续-->先创建 {} 然后根据下标，将当前的节点插入，再判断该节点是否有孩子节点。
#这样完成递归

在贴一个用队列的思路，官方题解

class Solution {
public:
    vector<vector<int>> levelOrder(Node* root) {
        if (!root) {
            return {};
        }

        vector<vector<int>> ans;
        queue<Node*> q;
        q.push(root);

        while (!q.empty()) {
            int cnt = q.size();
            vector<int> level;
            for (int i = 0; i < cnt; ++i) {
                Node* cur = q.front();
                q.pop();
                level.push_back(cur->val);
                for (Node* child: cur->children) {
                    q.push(child);
                }
            }
            ans.push_back(move(level));
        }

        return ans;
    }
};