[Programmers] x만큼 간격이 있는 n개의 숫자 (Stream 사용)

 

 

 

* 내 풀이 

 

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

        answer[0] = x;
        for(int i = 1; i < n; i++) {
            answer[i] = answer[i-1] + x;
        }

        return answer;
    }
}

 

 

* Stream 사용한 풀이 

 

import java.util.stream.LongStream;

class Solution {
    public long[] solution(int x, int n) {

        return LongStream.iterate(x, i-> x+i).limit(n).toArray();
    }
}

 

* Stream.iterate(초기값, 어떻게 변경될지).limit()

 

i를 갖고 돌리는데, 초기값 x에서 시작해서 i에다가 x를 더함