콜라츠 추측

1937년 Collatz란 사람에 의해 제기된 이 추측은, 주어진 수가 1이 될 때까지 다음 작업을 반복하면, 모든 수를 1로 만들 수 있다는 추측입니다. 작업은 다음과 같습니다.

• 1-1. 입력된 수가 짝수라면 2로 나눕니다.
• 1-2. 입력된 수가 홀수라면 3을 곱하고 1을 더합니다.
• 2. 결과로 나온 수에 같은 작업을 1이 될 때까지 반복합니다.
  • 예를 들어, 주어진 수가 6이라면 6 → 3 → 10 → 5 → 16 → 8 → 4 → 2 → 1 이 되어 총 8번 만에 1이 됩니다. 위 작업을 몇 번이나 반복해야 하는지 반환하는 함수, solution을 완성해 주세요. 단, 주어진 수가 1인 경우에는 0을, 작업을 500번 반복할 때까지 1이 되지 않는다면 –1을 반환해 주세요.
    문제링크

풀이
while을 쓸지 for을 쓸지 고민했는데 1이 되지 않을때까지 횟수를 세야해서 while ~ if ~ 로 표현

def solution(num):
    answer = 0 # 몇 번 계산했는지 세기
    
    while num != 1: #num이 1이 아닐때까지 반복
        
        if num % 2 == 0: # num이 짝수라면 
            num = num / 2 # 2로 나누기 
            
        else: # 홀수라면 
            num = num * 3 + 1 # 3num + 1
            
        answer += 1 # 계산할때마다 +1
        
    if answer > 500:
        answer = -1
    
    return answer

다른사람 풀이

def collatz(num):
    if num == 1:  # num이 처음부터 1이면 바로 0 반환
        return 0
    
    for i in range(500):
        num = num / 2 if num % 2 == 0 else num * 3 + 1
        if num == 1:
            return i + 1
    return -1

for 문으로 푼 사례가 있어 가져왔다
아주 간단해보이고 좋다 😭