mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-12-21 02:30:15 +00:00
bc8df6de31
* [pre-commit.ci] pre-commit autoupdate updates: - [github.com/astral-sh/ruff-pre-commit: v0.2.2 → v0.3.2](https://github.com/astral-sh/ruff-pre-commit/compare/v0.2.2...v0.3.2) - [github.com/pre-commit/mirrors-mypy: v1.8.0 → v1.9.0](https://github.com/pre-commit/mirrors-mypy/compare/v1.8.0...v1.9.0) * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
83 lines
2.0 KiB
Python
83 lines
2.0 KiB
Python
"""
|
|
Project Euler Problem 77: https://projecteuler.net/problem=77
|
|
|
|
It is possible to write ten as the sum of primes in exactly five different ways:
|
|
|
|
7 + 3
|
|
5 + 5
|
|
5 + 3 + 2
|
|
3 + 3 + 2 + 2
|
|
2 + 2 + 2 + 2 + 2
|
|
|
|
What is the first value which can be written as the sum of primes in over
|
|
five thousand different ways?
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from functools import lru_cache
|
|
from math import ceil
|
|
|
|
NUM_PRIMES = 100
|
|
|
|
primes = set(range(3, NUM_PRIMES, 2))
|
|
primes.add(2)
|
|
prime: int
|
|
|
|
for prime in range(3, ceil(NUM_PRIMES**0.5), 2):
|
|
if prime not in primes:
|
|
continue
|
|
primes.difference_update(set(range(prime * prime, NUM_PRIMES, prime)))
|
|
|
|
|
|
@lru_cache(maxsize=100)
|
|
def partition(number_to_partition: int) -> set[int]:
|
|
"""
|
|
Return a set of integers corresponding to unique prime partitions of n.
|
|
The unique prime partitions can be represented as unique prime decompositions,
|
|
e.g. (7+3) <-> 7*3 = 12, (3+3+2+2) = 3*3*2*2 = 36
|
|
>>> partition(10)
|
|
{32, 36, 21, 25, 30}
|
|
>>> partition(15)
|
|
{192, 160, 105, 44, 112, 243, 180, 150, 216, 26, 125, 126}
|
|
>>> len(partition(20))
|
|
26
|
|
"""
|
|
if number_to_partition < 0:
|
|
return set()
|
|
elif number_to_partition == 0:
|
|
return {1}
|
|
|
|
ret: set[int] = set()
|
|
prime: int
|
|
sub: int
|
|
|
|
for prime in primes:
|
|
if prime > number_to_partition:
|
|
continue
|
|
for sub in partition(number_to_partition - prime):
|
|
ret.add(sub * prime)
|
|
|
|
return ret
|
|
|
|
|
|
def solution(number_unique_partitions: int = 5000) -> int | None:
|
|
"""
|
|
Return the smallest integer that can be written as the sum of primes in over
|
|
m unique ways.
|
|
>>> solution(4)
|
|
10
|
|
>>> solution(500)
|
|
45
|
|
>>> solution(1000)
|
|
53
|
|
"""
|
|
for number_to_partition in range(1, NUM_PRIMES):
|
|
if len(partition(number_to_partition)) > number_unique_partitions:
|
|
return number_to_partition
|
|
return None
|
|
|
|
|
|
if __name__ == "__main__":
|
|
print(f"{solution() = }")
|