tp8 step 1 solution

4sh · Feb 14, 2025 · 1a790c2 · 1a790c2
1 parent 89da25b
commit 1a790c2
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/tp8/src/main/kotlin/fmt/kotlin/fundamentals/Tp8.kt b/tp8/src/main/kotlin/fmt/kotlin/fundamentals/Tp8.kt
@@ -2,5 +2,12 @@ package fmt.kotlin.fundamentals
 
 class Tp8 {
 
-    fun getFirstPrimeNumbers(nbToFind: Int) = emptyList()
+    fun getFirstPrimeNumbers(nbToFind: Int) = generateSequence(2) { it + 1 }
+        .filter { it.isPrime() }
+        .take(nbToFind)
+        .toList()
+
+    private fun Int.isPrime() = generateSequence(2) { it + 1 }
+        .takeWhile { it < this }
+        .none() { this % it == 0 }
 }