mirror of
https://github.com/correl/euler.git
synced 2024-11-27 11:09:54 +00:00
46 lines
1 KiB
Python
46 lines
1 KiB
Python
"""Find the 10001st prime.
|
|
|
|
By listing the first six prime numbers: 2, 3, 5, 7, 11, and 13, we can see that the 6th prime is 13.
|
|
What is the 10001st prime number?
|
|
"""
|
|
|
|
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
|
|
|
|
def main():
|
|
print '6th Prime', primes(6)[-1]
|
|
print '10001st Prime', primes(10001)[-1]
|
|
|
|
if __name__ == '__main__':
|
|
main()
|