Improve Project Euler problem 043 solution 1 (#5772)

* updating DIRECTORY.md

* Fix typo

* Improve solution

Co-authored-by: github-actions <${GITHUB_ACTOR}@users.noreply.github.com>
This commit is contained in:
Maxim Smolskiy
2021-11-05 22:44:24 +03:00
committed by GitHub
parent 48960268a2
commit 1a43c92c77
2 changed files with 17 additions and 7 deletions

View File

@ -33,9 +33,18 @@ def is_substring_divisible(num: tuple) -> bool:
>>> is_substring_divisible((1, 4, 0, 6, 3, 5, 7, 2, 8, 9))
True
"""
tests = [2, 3, 5, 7, 11, 13, 17]
if num[3] % 2 != 0:
return False
if (num[2] + num[3] + num[4]) % 3 != 0:
return False
if num[5] % 5 != 0:
return False
tests = [7, 11, 13, 17]
for i, test in enumerate(tests):
if (num[i + 1] * 100 + num[i + 2] * 10 + num[i + 3]) % test != 0:
if (num[i + 4] * 100 + num[i + 5] * 10 + num[i + 6]) % test != 0:
return False
return True
@ -43,17 +52,15 @@ def is_substring_divisible(num: tuple) -> bool:
def solution(n: int = 10) -> int:
"""
Returns the sum of all pandigital numbers which pass the
divisiility tests.
divisibility tests.
>>> solution(10)
16695334890
"""
list_nums = [
return sum(
int("".join(map(str, num)))
for num in permutations(range(n))
if is_substring_divisible(num)
]
return sum(list_nums)
)
if __name__ == "__main__":