mirror of
https://github.com/correl/euler.git
synced 2024-11-30 11:09:52 +00:00
37 lines
876 B
Python
37 lines
876 B
Python
import math
|
|
|
|
def is_prime(n):
|
|
if n == 1:
|
|
return False
|
|
if n < 4:
|
|
return True
|
|
if n % 2 == 0:
|
|
return False
|
|
if n < 9:
|
|
return True
|
|
if n % 3 == 0:
|
|
return False
|
|
|
|
r = math.floor(math.sqrt(n))
|
|
f = 5
|
|
while f <= r:
|
|
if (n % f == 0) or (n % (f + 2) == 0):
|
|
return False
|
|
f = f + 6
|
|
return True
|
|
|
|
def primes(max_count = 0, max_value = 0):
|
|
if not max_count and not max_value:
|
|
raise Exception('There must be a constraint on how many primes to return!')
|
|
primes = [2]
|
|
i = 1
|
|
while (not max_count or len(primes) < max_count) \
|
|
and (not max_value or i < max_value):
|
|
i = i + 2
|
|
if is_prime(i):
|
|
primes.append(i)
|
|
return primes
|
|
|
|
if __name__ == '__main__':
|
|
print '6th Prime', primes(6)[-1]
|
|
print '10001st Prime', primes(10001)[-1]
|