[코딩테스트 Lv.0] 배열 만들기 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 합니다.

---

내 풀이

Java

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

C

#include <stdio.h>
#include <stdbool.h>
#include <stdlib.h>

int* solution(int n, int k) {
    // return 값은 malloc 등 동적 할당을 사용해주세요. 할당 길이는 상황에 맞게 변경해주세요.
    int* answer = (int*)malloc(sizeof(int) * (n / k));
    int i, j;
    for(i = k, j = 0; i <= n; i++){
        if(i % k == 0)
            answer[j++] = i;
    }
    return answer;
}

Python

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

---

https://school.programmers.co.kr/learn/courses/30/lessons/181901

프로그래머스

 

---

[코딩테스트 Lv.0] 배열 만들기 3

 

[코딩테스트 Lv.0] 배열 만들기 6