Create n-ary-tree-postorder-traversal.cpp

Author: kamyu104

C++/n-ary-tree-postorder-traversal.cpp

@@ -0,0 +1,39 @@
+// Time:  O(n)
+// Space: O(h)
+
+/*
+// Definition for a Node.
+class Node {
+public:
+    int val;
+    vector<Node*> children;
+
+    Node() {}
+
+    Node(int _val, vector<Node*> _children) {
+        val = _val;
+        children = _children;
+    }
+};
+*/
+class Solution {
+public:
+    vector<int> postorder(Node* root) {
+        if (!root) {
+            return {};
+        }
+        vector<int> result;
+        vector<Node*> stack{root};
+        while (!stack.empty()) {
+            auto node = stack.back(); stack.pop_back();
+            result.emplace_back(node->val);
+            for (const auto& child : node->children) {
+                if (child) {
+                    stack.emplace_back(child);
+                }
+            }
+        }
+        reverse(result.begin(), result.end());
+        return result;
+    }
+};