1
0
Fork 0
mirror of https://gitlab.com/mfocko/LeetCode.git synced 2024-11-09 15:59:06 +01:00

problems: add n-ary tree level-order traversal

Signed-off-by: Matej Focko <mfocko@redhat.com>
This commit is contained in:
Matej Focko 2022-07-23 13:10:25 +02:00
parent 6c3cfcd876
commit 7c3f9ba87e
Signed by: mfocko
GPG key ID: 7C47D46246790496

View file

@ -0,0 +1,47 @@
/*
// 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 {
void levelOrder(vector<vector<int>>& traversal, Node* root, int level)
{
if (root == nullptr) {
return;
}
if (level >= traversal.size()) {
traversal.push_back({});
}
traversal[level].push_back(root->val);
for (auto child : root->children) {
levelOrder(traversal, child, level + 1);
}
}
public:
vector<vector<int>> levelOrder(Node* root)
{
vector<vector<int>> result;
levelOrder(result, root, 0);
return result;
}
};