LeetCode_429.N叉树的层序遍历

题目:

给定一个 N 叉树,返回其节点值的层序遍历。 (即从左到右,逐层遍历)。

例如,给定一个 3叉树 :

在这里插入图片描述

返回其层序遍历:

[
     [1],
     [3,2,4],
     [5,6]
]

说明:

树的深度不会超过 1000。
树的节点总数不会超过 5000。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/n-ary-tree-level-order-traversal
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

方法一:迭代法:

解题思路:
提到层序遍历,最普遍的方法就是树的广度优先遍历。这道题的难点在于如何按层输出,方法是在常规的广度优先遍历的while循环中再加入一层for循环,for循环的大小是当前的queue.size,在访问到树的第n层时就把第(n+1)层的子结点全部插入到队列queue中,则下次while循环是queue的大小就是第(n+1)层的所有结点,插入到一个临时vector< int > layer中即可。

时间复杂度: O(n) //每个结点遍历一次
空间复杂度: O(n) //queue

/*
// 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 {
    
    
private:
    vector<vector<int>> result;
public:
    vector<vector<int>> levelOrder(Node* root) {
    
    
        queue<Node*> q;
        if(root != nullptr) q.push(root);
        while(!q.empty()) {
    
    
            int size = q.size();
            vector<int> layer;
            for(int i = 0; i < size; ++i) {
    
     //第 n 层的结点全部插入vector
                layer.push_back(q.front()->val);
                for(Node* node_ : q.front()->children)  //第 n+1 层的结点全部入队queue
                    q.push(node_);
                q.pop();
            }
            result.push_back(layer);
        }
        return result;
    }
};

猜你喜欢

转载自blog.csdn.net/ArtAndLife/article/details/109014719