problems: add n-ary tree preorder traversal
Signed-off-by: Matej Focko <mfocko@redhat.com>
This commit is contained in:
parent
7c3f9ba87e
commit
b59f949d8f
1 changed files with 43 additions and 0 deletions
43
problems/n-ary-tree-preorder-traversal.cpp
Normal file
43
problems/n-ary-tree-preorder-traversal.cpp
Normal file
|
@ -0,0 +1,43 @@
|
||||||
|
/*
|
||||||
|
// 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 preorder(vector<int>& traversal, Node* root)
|
||||||
|
{
|
||||||
|
if (root == nullptr) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
traversal.push_back(root->val);
|
||||||
|
for (auto child : root->children) {
|
||||||
|
preorder(traversal, child);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public:
|
||||||
|
vector<int> preorder(Node* root)
|
||||||
|
{
|
||||||
|
vector<int> result;
|
||||||
|
preorder(result, root);
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
};
|
Loading…
Reference in a new issue