Python

프로그래머스 - 콜라츠 추측

Wix 2024. 4. 4. 10:00

문제

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

해설

def solution(num):
    answer = num
    count = 0
    for _ in range(500):
      if answer == 1:
        break
      if answer % 2 == 0:
        answer /= 2
      else:
        answer = answer * 3 + 1
      count += 1
      
    return count if answer == 1 else -1

 

설명에 나온대로, 짝수면 2로 나누고 홀수면 3을 곱하고 1을 더해주는 작업을 해주면서 count를 계산한다.

만약 500번을 다 실행했는데도 answer가 1이 아니면 -1을 반환하고 answer가 1이면 count를 반환한다.