mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-12-24 12:10:16 +00:00
1f8a21d727
* Tighten up psf/black and flake8
* Fix some tests
* Fix some E741
* Fix some E741
* updating DIRECTORY.md
Co-authored-by: github-actions <${GITHUB_ACTOR}@users.noreply.github.com>
45 lines
1.2 KiB
Python
45 lines
1.2 KiB
Python
from typing import List
|
|
|
|
|
|
def collatz_sequence(n: int) -> List[int]:
|
|
"""
|
|
Collatz conjecture: start with any positive integer n. The next term is
|
|
obtained as follows:
|
|
If n term is even, the next term is: n / 2 .
|
|
If n is odd, the next term is: 3 * n + 1.
|
|
|
|
The conjecture states the sequence will always reach 1 for any starting value n.
|
|
Example:
|
|
>>> collatz_sequence(2.1)
|
|
Traceback (most recent call last):
|
|
...
|
|
Exception: Sequence only defined for natural numbers
|
|
>>> collatz_sequence(0)
|
|
Traceback (most recent call last):
|
|
...
|
|
Exception: Sequence only defined for natural numbers
|
|
>>> collatz_sequence(43) # doctest: +NORMALIZE_WHITESPACE
|
|
[43, 130, 65, 196, 98, 49, 148, 74, 37, 112, 56, 28, 14, 7,
|
|
22, 11, 34, 17, 52, 26, 13, 40, 20, 10, 5, 16, 8, 4, 2, 1]
|
|
"""
|
|
|
|
if not isinstance(n, int) or n < 1:
|
|
raise Exception("Sequence only defined for natural numbers")
|
|
|
|
sequence = [n]
|
|
while n != 1:
|
|
n = 3 * n + 1 if n & 1 else n // 2
|
|
sequence.append(n)
|
|
return sequence
|
|
|
|
|
|
def main():
|
|
n = 43
|
|
sequence = collatz_sequence(n)
|
|
print(sequence)
|
|
print(f"collatz sequence from {n} took {len(sequence)} steps.")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|