LC 2649. Nested Array Generator (JS)

Author: raul-sauco

README.md

@@ -529,6 +529,7 @@ Solutions to LeetCode problems. The first column links to the problem in LeetCod
 | [2636. Promise Pool][lc2636]                                                                  | 🟠 Medium  | [![js](res/js.png)][lc2636js]                                                                         |
 | [2637. Promise Time Limit][lc2637]                                                            | 🟢 Easy    | [![js](res/js.png)][lc2637js]                                                                         |
 | [2648. Generate Fibonacci Sequence][lc2648]                                                   | 🟢 Easy    | [![js](res/js.png)][lc2648js]                                                                         |
+| [2649. Nested Array Generator][lc2649]                                                        | 🟠 Medium  | [![js](res/js.png)][lc2649js]                                                                         |
 | [2665. Counter II][lc2665]                                                                    | 🟢 Easy    | [![js](res/js.png)][lc2665js]                                                                         |
 | [2666. Allow One Function Call][lc2666]                                                       | 🟢 Easy    | [![js](res/js.png)][lc2666js]                                                                         |
 | [2667. Create Hello World Function][lc2667]                                                   | 🟢 Easy    | [![js](res/js.png)][lc2667js]                                                                         |
@@ -1686,6 +1687,8 @@ Solutions to LeetCode problems. The first column links to the problem in LeetCod
 [lc2637js]: leetcode/promise-time-limit.js
 [lc2648]: https://leetcode.com/problems/generate-fibonacci-sequence/
 [lc2648js]: leetcode/generate-fibonacci-sequence.js
+[lc2649]: https://leetcode.com/problems/nested-array-generator/
+[lc2649js]: leetcode/nested-array-generator.js
 [lc2665]: https://leetcode.com/problems/counter-ii/
 [lc2665js]: leetcode/counter-ii.js
 [lc2666]: https://leetcode.com/problems/allow-one-function-call/

leetcode/nested-array-generator.js

@@ -0,0 +1,35 @@
+// 2649. Nested Array Generator
+// 🟠 Medium
+//
+// https://leetcode.com/problems/nested-array-generator/
+//
+// Tags: Javascript
+
+// A generator that returns values of a nested array as if produced by an
+// inorder traversal.
+//
+// Time complexity: O(n) - Each call takes O(1), n calls will take O(n)
+// Space complexity: O(n) - The call stack could grow to size n.
+//
+// Runtime 184 ms Beats 73.7%
+// Memory 74 MB Beats 50.62%
+/**
+ * @param {Array} arr
+ * @return {Generator}
+ */
+var inorderTraversal = function* (arr) {
+  if (!Array.isArray(arr)) {
+    yield arr;
+    return;
+  }
+  for (let i = 0; i < arr.length; i++) {
+    yield* inorderTraversal(arr[i]);
+  }
+};
+
+/**
+ * const gen = inorderTraversal([1, [2, 3]]);
+ * gen.next().value; // 1
+ * gen.next().value; // 2
+ * gen.next().value; // 3
+ */