[JAVA 문제 풀이] 141. 배열 만들기 1

내가 작성한 코드

class Solution {
    public int[] solution(int n, int k) {
        int[] answer = new int[n / k];

        for(int i = 1; i <= answer.length; i++ ){
            answer[i - 1] = i * k;
        }
        
        return answer;
    }
}

다른 사람의 코드

import java.util.stream.IntStream;

class Solution {
    public int[] solution(int n, int k) {
        return IntStream.rangeClosed(1, n).filter(i -> i % k == 0).toArray();
    }
}

• 스트림 사용