[알고리즘문제풀기] 배열 만들기 1

문제

내가 작성한 정답

class Solution {
    public int[] solution(int n, int k) {
        int[] answer = new int[n/k];
        int idx = 0;
        for(int i =1; i<=n; i++){
            if(i%k==0) answer[idx++] = i;
        }
        return answer;
    }
}

다른 사람들의 정답

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


        int[] answer = new int[count];

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

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