N-ary Tree Level Order Traversal

Posted 2021-03-05 liuqiujie

/*
// 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) {
        queue<Node*> q;
        vector<vector<int>> vec;
        if(!root) return vec;
        q.push(root);
        while(!q.empty()) {
            vector<int> tmp;
            for(int size=q.size();size;--size) {
                Node* node=q.front();q.pop();
                tmp.push_back(node->val);
                for(auto& child:node->children) q.push(child);
            }
            vec.push_back(tmp);
        }
        return vec;
    }
};