[프로그래머스 기초_java] 배열 만들기 1

 Solution 1) 내 풀이

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

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

        for (int i = 1; i <= n; i++) {
            if (i % k == 0) {
                list.add(i);
            }
        }
        return list.stream().mapToInt(Integer::intValue).toArray();
    }
}

저번에 썼던 list to array에 필요한 stream을 써봤다.

 

Solution 2) 다른 사람의 풀이

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

나는 배열의 크기를 미리 지정할 수 없을 것 같아 list를 썼었으나, 미리 알수 있는 방법이 있었고
덕분에 코드가 더 간결해진 모습을 볼 수 있다.
참고 하자. (참꼬도 하자..)