mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-12-24 12:10:16 +00:00
83b8b3c141
* Fixes: #3869: Optimize Project Euler's Problem 74's Solution 2 * updating DIRECTORY.md Co-authored-by: github-actions <${GITHUB_ACTOR}@users.noreply.github.com>
127 lines
3.5 KiB
Python
127 lines
3.5 KiB
Python
"""
|
|
Project Euler Problem 074: https://projecteuler.net/problem=74
|
|
|
|
Starting from any positive integer number
|
|
it is possible to attain another one summing the factorial of its digits.
|
|
|
|
Repeating this step, we can build chains of numbers.
|
|
It is not difficult to prove that EVERY starting number
|
|
will eventually get stuck in a loop.
|
|
|
|
The request is to find how many numbers less than one million
|
|
produce a chain with exactly 60 non repeating items.
|
|
|
|
Solution approach:
|
|
This solution simply consists in a loop that generates
|
|
the chains of non repeating items.
|
|
The generation of the chain stops before a repeating item
|
|
or if the size of the chain is greater then the desired one.
|
|
After generating each chain, the length is checked and the
|
|
counter increases.
|
|
"""
|
|
|
|
factorial_cache = {}
|
|
factorial_sum_cache = {}
|
|
|
|
|
|
def factorial(a: int) -> int:
|
|
"""Returns the factorial of the input a
|
|
>>> factorial(5)
|
|
120
|
|
|
|
>>> factorial(6)
|
|
720
|
|
|
|
>>> factorial(0)
|
|
1
|
|
"""
|
|
|
|
# The factorial function is not defined for negative numbers
|
|
if a < 0:
|
|
raise ValueError("Invalid negative input!", a)
|
|
|
|
if a in factorial_cache:
|
|
return factorial_cache[a]
|
|
|
|
# The case of 0! is handled separately
|
|
if a == 0:
|
|
factorial_cache[a] = 1
|
|
else:
|
|
# use a temporary support variable to store the computation
|
|
temporary_number = a
|
|
temporary_computation = 1
|
|
|
|
while temporary_number > 0:
|
|
temporary_computation *= temporary_number
|
|
temporary_number -= 1
|
|
|
|
factorial_cache[a] = temporary_computation
|
|
return factorial_cache[a]
|
|
|
|
|
|
def factorial_sum(a: int) -> int:
|
|
"""Function to perform the sum of the factorial
|
|
of all the digits in a
|
|
|
|
>>> factorial_sum(69)
|
|
363600
|
|
"""
|
|
if a in factorial_sum_cache:
|
|
return factorial_sum_cache[a]
|
|
# Prepare a variable to hold the computation
|
|
fact_sum = 0
|
|
|
|
""" Convert a in string to iterate on its digits
|
|
convert the digit back into an int
|
|
and add its factorial to fact_sum.
|
|
"""
|
|
for i in str(a):
|
|
fact_sum += factorial(int(i))
|
|
factorial_sum_cache[a] = fact_sum
|
|
return fact_sum
|
|
|
|
|
|
def solution(chain_length: int = 60, number_limit: int = 1000000) -> int:
|
|
"""Returns the number of numbers that produce
|
|
chains with exactly 60 non repeating elements.
|
|
>>> solution(10, 1000)
|
|
26
|
|
"""
|
|
|
|
# the counter for the chains with the exact desired length
|
|
chain_counter = 0
|
|
|
|
for i in range(1, number_limit + 1):
|
|
|
|
# The temporary list will contain the elements of the chain
|
|
chain_set = {i}
|
|
len_chain_set = 1
|
|
last_chain_element = i
|
|
|
|
# The new element of the chain
|
|
new_chain_element = factorial_sum(last_chain_element)
|
|
|
|
# Stop computing the chain when you find a repeating item
|
|
# or the length it greater then the desired one.
|
|
|
|
while new_chain_element not in chain_set and len_chain_set <= chain_length:
|
|
chain_set.add(new_chain_element)
|
|
|
|
len_chain_set += 1
|
|
last_chain_element = new_chain_element
|
|
new_chain_element = factorial_sum(last_chain_element)
|
|
|
|
# If the while exited because the chain list contains the exact amount
|
|
# of elements increase the counter
|
|
if len_chain_set == chain_length:
|
|
chain_counter += 1
|
|
|
|
return chain_counter
|
|
|
|
|
|
if __name__ == "__main__":
|
|
import doctest
|
|
|
|
doctest.testmod()
|
|
print(f"{solution()}")
|