34. 콜라츠 수열 만들기

school.programmers.co.kr

https://school.programmers.co.kr/learn/courses/30/lessons/181919

내 답안

import java.util.ArrayList;
import java.util.List;

class Solution {
    public int[] solution(int n) {
        List<Integer> list = new ArrayList<>();
        list.add(n);

        while (n != 1) {
            if (n % 2 == 0) {
                n /= 2;
            } else {
                n = 3 * n + 1;
            }
            list.add(n);
        }

        int[] arr = new int[list.size()];
        for (int i = 0; i < list.size(); i++) {
            arr[i] = list.get(i);
        }
        return arr;
    }
}
Java
복사

다른 사람 풀이법

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();
    }
}
Java
복사

import java.util.LinkedList;
import java.util.Queue;

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

import java.util.*;
import java.util.stream.Stream;

class Solution {
    public int[] solution(int n) {
        return Stream.iterate(n, i -> i >= 1, i -> i == 1 ? 0 : i % 2 == 0 ? i / 2 : 3 * i + 1).mapToInt(Integer::intValue).toArray();
    }
}
Java
복사

import java.util.stream.*;

class Solution {
    public int[] solution(int start, int end) {
        return IntStream.rangeClosed(start, end).toArray();
    }
}
Java
복사

[java] IntStream.iterate() 와 Stream.iterate() 의 차이점은?

안녕하세요. 궁금증연구소입니다. 오늘 포스팅 주제는 "[java] IntStream.iterate()와 Stream.iterate()의 차이점은?"입니다. 프로그래머스의 콜라츠 수열 만들기를 풀고, 다른 사람의 풀이를 봤는데 이상한 점이 있어 포스팅을 해봅니다. 먼저 위 풀이를 보면, IntStream의 iterate를 활용해서 1보다 큰 값인 경우에 반복하도록 Stream을 만들고, IntStream concat을 활용해서 스트림 내에 1을 따로 추가해 주었습니다. 그 이유는 IntStream.iterate()는 두 번째 파라매터인 조건식이 false를 리턴하자마자 stream이 끝나기 때문입니다. 위 코드에서 값의 흐름을 보면 2까지만 스트림에 포함되는 것을 알 수 있습니다. 왜냐하면 2 가 1이 된..

https://zorba10004.tistory.com/401

Stream.iterate() 과 IntStream.iterate() 에 의미있는 차이점은 없고 i 가 1 일 때 0 으로 보내서 종결시키는 조건이 핵심