mathdatech/Python
1
0
Fork 0
mirror of https://github.com/TheAlgorithms/Python.git synced 2024-12-18 01:00:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

Change prime_sieve_eratosthenes.py to return list (#8062)

Browse Source
This commit is contained in:
Tianyi Zheng 2023-01-01 17:10:59 -08:00 committed by GitHub
parent d29afca93b
commit 7c1d23d448
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 22 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
maths/prime_sieve_eratosthenes.py
View File

@ -12,34 +12,43 @@ https://en.wikipedia.org/wiki/Sieve_of_Eratosthenes
""" """
def prime_sieve_eratosthenes(num): def prime_sieve_eratosthenes(num: int) -> list[int]:
""" """
print the prime numbers up to n Print the prime numbers up to n
>>> prime_sieve_eratosthenes(10) >>> prime_sieve_eratosthenes(10)
2,3,5,7, [2, 3, 5, 7]
>>> prime_sieve_eratosthenes(20) >>> prime_sieve_eratosthenes(20)
2,3,5,7,11,13,17,19, [2, 3, 5, 7, 11, 13, 17, 19]
>>> prime_sieve_eratosthenes(2)
[2]
>>> prime_sieve_eratosthenes(1)
[]
>>> prime_sieve_eratosthenes(-1)
Traceback (most recent call last):
...
ValueError: Input must be a positive integer
""" """
primes = [True for i in range(num + 1)] if num <= 0:
p = 2 raise ValueError("Input must be a positive integer")
primes = [True] * (num + 1)
p = 2
while p * p <= num: while p * p <= num:
if primes[p]: if primes[p]:
for i in range(p * p, num + 1, p): for i in range(p * p, num + 1, p):
primes[i] = False primes[i] = False
p += 1 p += 1
for prime in range(2, num + 1): return [prime for prime in range(2, num + 1) if primes[prime]]
if primes[prime]:
print(prime, end=",")
if __name__ == "__main__": if __name__ == "__main__":
import doctest import doctest
doctest.testmod() doctest.testmod()
num = int(input())
prime_sieve_eratosthenes(num) user_num = int(input("Enter a positive integer: ").strip())
print(prime_sieve_eratosthenes(user_num))