File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed
project_euler/problem_060 Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -175,17 +175,17 @@ def add_prime(primes: list[int]) -> list[int]:
175175 return primes
176176
177177
178- def generate_primes (n : int , flag : int = 1 ) -> list [int ]:
178+ def generate_primes (num_primes : int , flag : int = 1 ) -> list [int ]:
179179 """
180- Ensure we have at least n primes in the selected list .
180+ Generates a list of the first num_primes primes based on their modulo 3 value .
181181
182182 >>> generate_primes(5, 1)
183183 [3, 7, 13, 19, 31]
184184 >>> generate_primes(5, 2)
185185 [3, 5, 11, 17, 23]
186186 """
187187 primes = prime_mod_3_is_1_list if flag == 1 else prime_mod_3_is_2_list
188- while len (primes ) < n :
188+ while len (primes ) < num_primes :
189189 primes = add_prime (primes )
190190 return primes
191191
@@ -212,7 +212,7 @@ def solution(
212212 raise ValueError ("flag must be either 1 or 2" )
213213 primes = generate_primes (prime_limit , flag )
214214
215- def search (chain ) :
215+ def search (chain : tuple ) -> tuple [ int , ...] | None :
216216 """
217217 Recursive backtracking search to find a valid set of primes.
218218 A threshold is used to ensure we don't exceed the smallest sum.
You can’t perform that action at this time.
0 commit comments