[프로그래머스] 콜라츠 수열 만들기

🔍 문제 설명

---

 

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

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

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

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

🚨 제한 사항

---

1 ≤ n ≤ 1,000

 

 

🗒️ 입출력 예

---

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

 

💡 접근 방법

---

 

n값을 계속 계산해가며 배열에 넣어주어야 하기 때문에 ArrayList를 사용한다. (배열의 크기가 가변적)

 

1. n이 1보다 크면 짝수인지 홀수인지를 구분한다. (while문 사용)

2. while문을 통해 n이 1과 같아지면, 반복문을 빠져나온다. 그 뒤 1을 더해준다.

3. ArrayLlist를 int형 배열로 변환해준다.

 

 

import java.util.ArrayList;
import java.util.stream.*;

class Solution {
    public int[] solution(int n) {
        ArrayList<Integer> answer = new ArrayList<>();
        while (n > 1) {
            answer.add(n);
            if (n % 2 == 0) n = n / 2;
            else n = n * 3 + 1;
        }
        answer.add(1);
        return answer.stream().mapToInt(i -> i).toArray();
    }
}

 

💚 다른 사람의 풀이

---

import java.util.stream.IntStream;

class Solution {
    public int[] solution(int n) {
        return IntStream.concat(
                        IntStream.iterate(n, i -> i > 1, i -> i % 2 == 0 ? i / 2 : i * 3 + 1),
                        IntStream.of(1))
                .toArray();
    }
}

 

나는 stream 잘 다루는 사람 볼 때마다 리스펙한다... IntStream 이용해서 풀이하셨다. 한줄로 푸신 분도 계시던데 대박..