본문 바로가기

알고리즘 공부/프로그래머스 > Python3

[프로그래머스] 콜라츠 추측

https://programmers.co.kr/learn/courses/30/lessons/12943

 

코딩테스트 연습 - 콜라츠 추측

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

programmers.co.kr

 

[정답]

def solution(num):
    answer = 0
    for i in range(1,501):
        if num==1:
            break
        if num%2==0:
            num = num/2
            answer+=1
        else:
            num = num*3+1
            answer+=1
    if num!=1:
        answer = -1
    return answer


<실패 후기>

코드를 다 작성하고 '코드 채점하고 제출'을 하였는데, 나머지는 다 통과인데 한 개의 테스트에서 실패가 떴다.

왜 그런지 문제를 다시 읽어보았다. 제한사항에 입력된 수 num은 1이상 8000000 미만인 정수입니다.

만약 num의 입력값이 1이면, answer값은 바로 0이 되는 것이다.

원래 if num==1: break 부분을 num이 짝수/홀수일 때 계산을 한 뒤에 넣어주었는데,

num의 입력값이 1일 경우를 대비하여, num의 계산 앞에 위치시켜주었다.

그랬더니 테스트를 다 통과하고 정답이 되었다!