[Java Algorithm] 프로그래머스 Lv.0 _ 콜라츠 수열 만들기

[문제]

모든 자연수 x에 대해서 현재 값이 x이면 x가 짝수일 때는 2로 나누고, x가 홀수일 때는 3 * x + 1로 바꾸는 계산을 계속해서 반복하면 언젠가는 반드시 x가 1이 되는지 묻는 문제를 콜라츠 문제라고 부릅니다.
그리고 위 과정에서 거쳐간 모든 수를 기록한 수열을 콜라츠 수열이라고 부릅니다.
계산 결과 1,000 보다 작거나 같은 수에 대해서는 전부 언젠가 1에 도달한다는 것이 알려져 있습니다.
임의의 1,000 보다 작거나 같은 양의 정수 n이 주어질 때 초기값이 n인 콜라츠 수열을 return 하는 solution 함수를 완성해 주세요.

 

[Algorithm]

반복문(while)을 사용해 조건에 부합하는 경우 로직을 실행하도록 한다.

문제에서 제시한 분기처리를 if문을 통해 적절히 수행하고 while문을 빠져나온 뒤 결과를 return !

 

문제에서는 주어진 수 n이 1이 될 때까지 로직을 실행하도록 만들어야했기 때문에 

while문의 조건에 n != 1을 넣어주고,

그 내부 분기처리는 짝수와 홀수인 경우로 나눠서 알맞은 식을 넣어주었다.

 

콜라츠 수열은 1을 포함하여 수열을 반환해야 하기 때문에 반목문이 끝난 뒤 배열에 1을 추가해주도록 코드를 작성했다.

 

[Code]

import java.util.*;

class Solution {
        public int[] solution(int n) {
        List<Integer> result = new ArrayList<>();
        
        while (n != 1) { 
            result.add(n);
            if (n % 2 == 0) {
                n = n / 2;
            } else {
                n = 3 * n + 1;
            }
        }
        result.add(1);

        return result.stream().mapToInt(i -> i).toArray();
    }
        
}

 

[+ Plus]

* Java Stream _ mapToInt

mapToInt는 객체 타입을 기본 타입으로 바꿔주는 메서드

 

현재 List<Integer>로 선언되어 있는 result를 int[] 형식으로 반환하려면,

mapToInt를 통해 기본 객체로 바꿔주고 toArray()를 통해 배열로 변환해줘야 함.