aboutsummaryrefslogtreecommitdiff
path: root/challenge-158/deadmarshal/lua/ch-2.lua
blob: f3d0b9c25c58da68129a2d35e879767a7091e711 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
function is_prime(n)
   assert(type(n) == "number", "n must be a number!")
   if n <= 1 then return false end
   for i=2, math.sqrt(n) do
      if n % i == 0 then return false end
   end
   return true
end

function cuban_primes(n)
   assert(type(n) == "number", "n must be a number!")
   local t, i = {}, 1
   while i do
      local p = 3 * i * (i + 1) + 1
      if is_prime(p) then t[#t+1] = p end
      if p >= n then break end
      i = i + 1
   end
   return t
end

local t = cuban_primes(1000)
for i=1, #t do io.write(t[i], ' ') end