ZigzagGenerator,ZigzagIterator

Author: masx200

leetcode-test.iml

@@ -4,6 +4,7 @@
     <content url="file://$MODULE_DIR$" dumb="true">
       <sourceFolder url="file://$MODULE_DIR$/test" isTestSource="true" packagePrefix="com.github.masx200.leetcode_test" />
       <sourceFolder url="file://$MODULE_DIR$" isTestSource="false" packagePrefix="com.github.masx200.leetcode_test" />
+      <sourceFolder url="file://$MODULE_DIR$/zigzag-iterator" isTestSource="false" packagePrefix="com.github.masx200.leetcode_test.zigzag_iterator" />
       <excludeFolder url="file://$MODULE_DIR$/target" />
     </content>
   </component>

zigzag-iterator/ZigzagIterator.kt

@@ -0,0 +1,35 @@
+package com.github.masx200.leetcode_test.com.github.leetcode_test.zigzag_iterator
+
+
+private fun ZigzagGenerator(v1: List<Int>, v2: List<Int>): Sequence<Int> = sequence<Int> {
+    if (v1.size < v2.size) {
+        for (i in 0 until v1.size) {
+            yield(v1[i])
+
+            yield(v2[i])
+        }
+        for (i in v1.size until v2.size) {
+            yield(v2[i])
+
+        }
+    } else {
+        yieldAll(ZigzagGenerator(v2, v1))
+    }
+
+}
+
+class ZigzagIterator(v1: List<Int>, v2: List<Int>) {
+
+
+    private val generator = ZigzagGenerator(v1, v2)
+
+
+    private val iterator = generator.iterator()
+    fun next(): Int {
+        return iterator.next()
+    }
+
+    fun hasNext(): Boolean {
+        return iterator.hasNext()
+    }
+}