Skip to content

Commit 0fa14b6

Browse files
Update sol1.py
1 parent 55e4081 commit 0fa14b6

File tree

1 file changed

+11
-11
lines changed

1 file changed

+11
-11
lines changed

project_euler/problem_095/sol1.py

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,7 @@ def generate_primes(max_prime: int) -> list[int]:
6767
def multiply(
6868
chain: list[int],
6969
primes: list[int],
70-
prime: int,
70+
min_prime: int,
7171
prev_num: int,
7272
max_num: int,
7373
prev_sum: int,
@@ -78,30 +78,30 @@ def multiply(
7878
7979
>>> chain = [0] * 3
8080
>>> primes_degrees = {}
81-
>>> multiply(chain=chain, primes=[2], prime=2, prev_n=1, max_num=2, prev_sum=0, primes_degrees=primes_degrees)
81+
>>> multiply(chain=chain, primes=[2], min_prime=2, prev_n=1, max_num=2, prev_sum=0, primes_degrees=primes_degrees)
8282
>>> chain
8383
[0, 0, 0]
8484
>>> primes_degrees
8585
{}
8686
"""
8787

88-
num = prev_n * prime
89-
primes_degrees[prime] = primes_degrees.get(prime, 0) + 1
90-
if prev_num % prime != 0:
91-
new_sum = prev_sum * (prime + 1) + prev_num
88+
num = prev_num * min_prime
89+
primes_degrees[prime] = primes_degrees.get(min_prime, 0) + 1
90+
if prev_num % min_prime != 0:
91+
new_sum = prev_sum * (min_prime + 1) + prev_num
9292
else:
9393
new_sum = sum_primes(primes_degrees=primes_degrees, num=num)
9494
chain[num] = new_sum
9595
for prime in primes:
96-
if prime >= prime:
97-
num_n = p * num
96+
if prime >= min_prime:
97+
num_n = prime * num
9898
if num_n > max_num:
9999
break
100100
multiply(
101101
chain=chain,
102102
primes=primes,
103-
prime=prime,
104-
prev_n=num,
103+
min_prime=prime,
104+
prev_num=num,
105105
max_num=max_num,
106106
prev_sum=new_sum,
107107
primes_degrees=primes_degrees.copy(),
@@ -154,7 +154,7 @@ def solution(max_num: int = 1000000) -> int:
154154
multiply(
155155
chain=chain,
156156
primes=primes,
157-
prime=prime,
157+
min_prime=prime,
158158
prev_num=1,
159159
max_num=max_num,
160160
prev_sum=0,

0 commit comments

Comments
 (0)