[프로그래머스] Lv.0 배열 만들기1 (Python / Java)

 

코딩테스트 연습 - 배열 만들기 1 | 프로그래머스 스쿨 (programmers.co.kr)

프로그래머스

 

문제 설명

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

제한사항

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

입출력 예

 

---

 

문제풀이

 

[Python3]

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

 

[Java]

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