[프래그래머스] 콜라츠 추측(Python)

문제 설명

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을 반환해 주세요.

핵심 아이디어

1. while True문을 사용한다.

2. if와 elif를 적절히 사용한다. elif는 앞선 조건문 선상에서 다른 조건을 걸 때 쓰인다.

3. 값을 return 해주고, 꼭 break를 걸어준다.

첫 코드

def solution(num):
    answer = 0
    if num == 1:
        return 0
    while True:
        if num % 2 == 0:
            num = num // 2
            answer += 1
        elif num % 2 == 1:
            num = num * 3 + 1
            answer += 1
        if num == 1:
            return answer
            break
        if answer == 500:
            return -1
            break

개선된 코드

def solution(num):
    answer = 0
    
    while num != 1 :
        if answer > 500 :
            return -1
        if num % 2 == 0 :
            num = num/2
        elif num % 2 == 1 :
            num = num*3 + 1
        answer += 1
    
    return answer

처음 while문 조건에 1이 아니라는 조건을 걸면, 좀 더 간단하게 코드를 줄일 수 있었다.