Debug prime_functions

LivInTheLookingGlass · Sep 11, 2024 · 67cef94 · 67cef94
1 parent e0c152d
commit 67cef94
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/lua/src/lib/primes.lua b/lua/src/lib/primes.lua
@@ -49,8 +49,12 @@ local function prime_factors(n)
     local p = pgen.next()
 
     local function next()
-        print(p .. ", " .. n)
-        while p ~= nil and n % p ~= 0
+        if n == 1
+        then
+            return nil
+        end
+
+        while p and n % p ~= 0
         do
             p = pgen.next()
         end
@@ -66,7 +70,7 @@ local function prime_factors(n)
             return -1
         end
 
-        n = n / p
+        n = math.floor(n / p)
         return p
     end