[프로그래머스 코딩테스트] 콜라츠 수열 만들기(Java)

  문제 설명

  모든 자연수 x에 대해서 현재 값이 x이면 x가 짝수일 때는 2로 나누고, x가 홀수일 때는 3 * x + 1로 바꾸는 계산을 계속해서 반복하면 언젠가는 반드시 x가 1이 되는지 묻는 문제를 콜라츠 문제라고 부릅니다.

  그리고 위 과정에서 거쳐간 모든 수를 기록한 수열을 콜라츠 수열이라고 부릅니다.

  계산 결과 1,000 보다 작거나 같은 수에 대해서는 전부 언젠가 1에 도달한다는 것이 알려져 있습니다.

  임의의 1,000 보다 작거나 같은 양의 정수 n이 주어질 때 초기값이 n인 콜라츠 수열을 return 하는 solution 함수를 완성해 주세요.

 

  제한사항

• 1 <= n <= 1,000

 

  입출력 예

n	result
10	[10, 5, 16, 8, 4, 2, 1]

import java.util.ArrayList;
class Solution {
    public int[] solution(int n) {
        ArrayList<Integer> list = new ArrayList<>();
        while (n > 1) {
            list.add(n);
            if (n % 2 == 0) {
                n /= 2;
            } else {
                n = 3 * n + 1;
            }
        }
        list.add(1);
        
        int[] answer = new int[list.size()];
        int index = 0;
        for (int num: list) {
            answer[index++] = num;
        }
        return answer;
    }
}

  ArrayList를 생성합니다. 곧바로 answer 배열에 할당하면 좋겠지만, 배열의 크기가 불분명하기 때문입니다.

  while문은 n이 1보다 큰 동안 반복합니다. add 메소드로 list에 n을 추가합니다. 만약 n이 2로 나누어떨어진다면, 즉 짝수라면 n을 2로 나눕니다. 만약 홀수라면 n에 3 * n + 1을 할당합니다. 계산된 값이 다시 while문을 돌며 list에 추가됩니다. 이 과정을 반복합니다.

  while문이 1보다 큰 동안이므로 앞선 계산으로 추가된 list에 1은 포함되지 않습니다. 그러나 문제에서 항상 1에 도달한다고 했으므로 list에 원소 1을 추가합니다.

  answer 배열을 생성합니다. list를 answer에 복사할 것이므로 크기는 동일할 것입니다. answer의 원소 자리가 될 index 변수를 만듭니다. for-each문에서 list의 원소를 변수 num이 받습니다. answer의 배열 앞에서부터 list의 원소가 추가됩니다.