본문 바로가기
컴퓨터/알고리즘

[프로그래머스] Lv1 콜라츠 추측_python 문제 풀이

by 버니케이 2022. 7. 6.
반응형

출처

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

문제

주어진 수가 1이 될 때까지 다음 작업을 반복하면, 모든 수를 1로 만들 수 있다.
1-1. 입력된 수가 짝수라면 2로 나눕니다.
1-2. 입력된 수가 홀수라면 3을 곱하고 1을 더합니다.
2. 결과로 나온 수에 같은 작업을 1이 될 때까지 반복합니다.
주어진 수가 6이라면 6 → 3 → 10 → 5 → 16 → 8 → 4 → 2 → 1
위 작업을 몇 번이나 반복해야 하는지 반환하는 함수, solution을 완성하라

단 주어진 수가 1인 경우에는 0을, 작업을 500번 반복할 때까지 1이 되지 않는다면 –1

입출력예시

문제풀이1

def solution(num):
    count = 0

    while num != 1:
        if num % 2 == 0:
            num = num / 2
            count +=1
        elif num % 2 ==1:
            num = num * 3 + 1
            count +=1
        if count >500:
            return -1

    return count
반응형

댓글