[프로그래머스] Level 1 콜라츠 추측 - 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을 완성해 주세요. 단, 작업을 500번을 반복해도 1이 되지 않는다면 –1을 반환해 주세요.

제한 사항

• 입력된 수, num은 1 이상 8000000 미만인 정수입니다.

입출력 예

n	result
6	8
16	4
626331	-1

입출력 예 설명

입출력 예 #1
문제의 설명과 같습니다.

 

입출력 예 #2
16 -> 8 -> 4 -> 2 -> 1 이되어 총 4번만에 1이 됩니다.

 

입출력 예 #3
626331은 500번을 시도해도 1이 되지 못하므로 -1을 리턴해야 합니다.

풀이

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

어후 길다 길어...

1. num이 1일 경우엔 바로 0을 return 하도록 처음에 설정해줬다.
2. num이 1이 아닐 경우 다음과 같은 반복문을 거친다.
   1. 짝수일 경우, 2로 나누고 answer에 1을 더한다.
   2. 홀수일 경우, 3으로 곱한 값에 1을 더하고 answer에 1을 더한다.
3. answer가 500 이상일 경우, -1을 return 한다.

다른 사람 풀이

def collatz(num):
    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 loop를 500번까지 돌려주게끔 설정
• 3항 연산을 통해 더 짧은 code를 작성했다.