프로그래머스 181901번 배열 만들기 1 - Python, Java

문제 설명

프로그래머스 181901번 배열 만들기 1

정수 n과 k가 주어졌을 때, 1 이상 n이하의 정수 중에서 k의 배수를 오름차순으로 저장한 배열을 return 하는 solution 함수를 완성해 주세요.

제한 사항

• 1 ≤ n ≤ 1,000,000
• 1 ≤ k ≤ min(1,000, n)

입출력 예시

n	k	result
10	3	[3, 6, 9]
15	5	[5, 10, 15]

입출력 예 설명 1

1 이상 10 이하의 3의 배수는 3, 6, 9 이므로 [3, 6, 9]를 return 합니다.

입출력 예 설명 2

1 이상 15 이하의 5의 배수는 5, 10, 15 이므로 [5, 10, 15]를 return 합니다.

코드 구현

Python

def solution(n, k):
    answer = []
    
    for i in range(1, n + 1):
        if i % k == 0:
            answer.append(i)
    
    return answer

Java

import java.util.ArrayList;

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

        for (int i = 1; i <= n; i += 1) {
            if (i % k == 0) {
                answer.add(i);
            }
        }

        return answer.stream().mapToInt(i -> i).toArray();
    }
}