[JavaScript] 콜라츠 추측 (프로그래머스 Lv.1)

문제 설명

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을 완성해 주세요. 단, 주어진 수가 1인 경우에는 0을, 작업을 500번 반복할 때까지 1이 되지 않는다면 –1을 반환해 주세요.

제한 사항

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

입출력 예nresult

6	8
16	4
626331	-1

입출력 예 설명

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

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

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

 

 

 

문제 이해

- 어떤 수가 있고, 이 수를 나눠 1로 만들어야한다. 이를 무한 반복함.

- 만약 수가 홀수면 3을 곱하고 1을 더해서 짝수화 해야한다.

- 주어진 수가 1이라면 0을 리턴, 500번을 초과하면 -1을 리턴 => 횟수 체크 필요.

 

처음에 문제를 잘못 이해했는데, 3을 곱하고 1을 더하는 짝수화 과정을 초기에 한 번만 진행하는 줄 알았다..

생각해보면 아무리 짝수라도 2로 나누다보면 홀수가 나올 수 있음. 고로 짝수화 과정은 반복 중에 홀수면 계속 진행한다.

 

문제 풀이

- 입력 num이 1이면 0을 리턴

- 카운트 변수를 선언하고 초기 횟수를 1로 설정.

- while 반복문, 조건은 카운트 500이하동안으로.

- 반복문 내에선 삼항연산자를 통해 짝/홀을 판단하고 짝수일 경우 2를 나눔. 홀수라면 3을 곱하고 1을 더함.

- 그렇게 진행한 값이 1이라면 그대로 카운트 횟수를 리턴하고, 아니면 카운트 횟수를 1 증가시킨 뒤 다시 반복.

- 횟수가 500번을 초과하면 while문이 종료되므로 아래의 리턴 -1이 실행된다.

*포인트 : while문에서 return을 만나면 그대로 함수가 종료된다!

 

코드

function solution(num) {
    if (num===1) return 0
    let count = 1;
    
    while (count <= 500) {
        num = num%2===0 ? parseInt(num/2) : num*3 + 1
        
        if (num===1) return count
        else count += 1
    }
    return -1
}

 

while문 내의 마지막 구문에서도 삼항연산자를 써볼까 했는데 

return 삼항연산자식 으로 표현하기도 애매하고, 변수에 값을 넣는 과정도 아니라서 그냥 if문으로 적었다.