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

Lee·2022년 1월 18일
0

알고리즘

목록 보기
8/24

문제설명

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 미만인 정수입니다.

내가 쓴 코드

class Solution {
    public int solution(long num) { // 타입을 바꿔야 한다 (int -> long) 오버플로우로 오답
        int cnt = 0; // 콜라츠 추측 진행상황을 세는 카운터 변수
        
        while(cnt < 500) { // 500회 제한이 걸려 있으므로 반복문 사용
            if (num == 1) { // 숫자가 1이 되면 카운터 리턴
                return cnt;
            }
            
            if (num%2 == 0) { //숫자가 짝수일 경우 2로 나눈다
                num /= 2;
            } else {	// 숫자가 홀수일 경우 3을 곱하고 1을 더한다.
                num = (num * 3) + 1;
            }
            cnt++;
        }
        return -1;
    }
}

정리

콜라츠 추측을 몇번 적용했는지 확인하는 카운터 변수선언
whlie문을 사용하여 문제에서 제시한 500회 만큼 반복
콜라츠 추측을 진행한 숫자가 1이면 카운터 리턴
조건문을 사용해 num이 짝수인 경우와 홀수인 경우 구분
짝수인 경우 num을 2로 나눈다
홀수인 경우 num에 3을 곱하고 1을 더한다.
이 과정이 끝나면 카운터를 증가한다.

문제에서 요구한 사항으로 작성을 했을때 오답이 나와서 고생했다. 주어진대로 int를 return하게 되면 오버플로우로 초기화되면서 오답이 나오므로 long타입을 사용하였다.

profile
발전하고 싶은 백엔드 개발자

0개의 댓글