mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-11-30 16:31:08 +00:00
64 lines
1.9 KiB
Python
64 lines
1.9 KiB
Python
"""
|
|
Collatz conjecture: start with any positive integer n. Next term obtained from
|
|
the previous term as follows:
|
|
|
|
If the previous term is even, the next term is one half the previous term.
|
|
If the previous term is odd, the next term is 3 times the previous term plus 1.
|
|
The conjecture states the sequence will always reach 1 regardless of starting
|
|
n.
|
|
|
|
Problem Statement:
|
|
The following iterative sequence is defined for the set of positive integers:
|
|
|
|
n → n/2 (n is even)
|
|
n → 3n + 1 (n is odd)
|
|
|
|
Using the rule above and starting with 13, we generate the following sequence:
|
|
|
|
13 → 40 → 20 → 10 → 5 → 16 → 8 → 4 → 2 → 1
|
|
|
|
It can be seen that this sequence (starting at 13 and finishing at 1) contains
|
|
10 terms. Although it has not been proved yet (Collatz Problem), it is thought
|
|
that all starting numbers finish at 1.
|
|
|
|
Which starting number, under one million, produces the longest chain?
|
|
"""
|
|
|
|
|
|
def collatz_sequence(n):
|
|
"""Returns the Collatz sequence for n."""
|
|
sequence = [n]
|
|
while n != 1:
|
|
if n % 2 == 0:
|
|
n //= 2
|
|
else:
|
|
n = 3 * n + 1
|
|
sequence.append(n)
|
|
return sequence
|
|
|
|
|
|
def solution(n):
|
|
"""Returns the number under n that generates the longest Collatz sequence.
|
|
|
|
# The code below has been commented due to slow execution affecting Travis.
|
|
# >>> solution(1000000)
|
|
# {'counter': 525, 'largest_number': 837799}
|
|
>>> solution(200)
|
|
{'counter': 125, 'largest_number': 171}
|
|
>>> solution(5000)
|
|
{'counter': 238, 'largest_number': 3711}
|
|
>>> solution(15000)
|
|
{'counter': 276, 'largest_number': 13255}
|
|
"""
|
|
|
|
result = max([(len(collatz_sequence(i)), i) for i in range(1, n)])
|
|
return {"counter": result[0], "largest_number": result[1]}
|
|
|
|
|
|
if __name__ == "__main__":
|
|
result = solution(int(input().strip()))
|
|
print(
|
|
"Longest Collatz sequence under one million is %d with length %d"
|
|
% (result["largest_number"], result["counter"])
|
|
)
|