Skip to content

Commit 4896026

Browse files
Improve Project Euler Problem 10 Sol-1 (#5773)
* Improve Project Euler Problem 10 Sol-1 * Name correction * psf/black formatting * More formatting
1 parent 6b2b476 commit 4896026

File tree

1 file changed

+6
-15
lines changed

1 file changed

+6
-15
lines changed

Diff for: project_euler/problem_010/sol1.py

+6-15
Original file line numberDiff line numberDiff line change
@@ -28,11 +28,11 @@ def is_prime(n: int) -> bool:
2828
True
2929
"""
3030

31-
for i in range(2, int(sqrt(n)) + 1):
32-
if n % i == 0:
33-
return False
34-
35-
return True
31+
if 1 < n < 4:
32+
return True
33+
elif n < 2 or not n % 2:
34+
return False
35+
return not any(not n % i for i in range(3, int(sqrt(n) + 1), 2))
3636

3737

3838
def solution(n: int = 2000000) -> int:
@@ -49,16 +49,7 @@ def solution(n: int = 2000000) -> int:
4949
10
5050
"""
5151

52-
if n > 2:
53-
sum_of_primes = 2
54-
else:
55-
return 0
56-
57-
for i in range(3, n, 2):
58-
if is_prime(i):
59-
sum_of_primes += i
60-
61-
return sum_of_primes
52+
return sum(num for num in range(3, n, 2) if is_prime(num)) + 2 if n > 2 else 0
6253

6354

6455
if __name__ == "__main__":

0 commit comments

Comments
 (0)