Post

[Programmers] #181901 - 배열 만들기 1 [Java][C++][Python]

1 이상 n 이하의 정수 중 k의 배수를 오름차순으로 담은 배열을 반환하는 워밍업 문제.

[Programmers] #181901 - 배열 만들기 1 [Java][C++][Python]

문제 링크


1. 아이디어

k부터 시작해 k씩 더해가며 n을 넘지 않을 때까지의 값을 배열에 담으면, k의 배수가 이미 오름차순으로 얻어진다. 배수의 개수는 n / k개다.


2. 복잡도

접근시간공간
풀이$O(N / K)$$O(N / K)$

($N$ = 입력값 n, $K$ = 입력값 k)


3. 코드

풀이 [Java][C++][Python]

1
2
3
4
5
6
7
8
9
10
11
12
class Solution {
    public int[] solution(int n, int k) {
        int[] arr = new int[n / k];
        int idx = 0;

        for (int i = k; i <= n; i += k) {
            arr[idx++] = i;
        }

        return arr;
    }
}
1
2
3
4
5
6
7
8
9
10
11
#include <bits/stdc++.h>
using namespace std;

vector<int> solution(int n, int k) {
    vector<int> v;
    for (int i = k; i <= n; i += k) {
        v.push_back(i);
    }

    return v;
}
1
2
def solution(n, k):
    return list(range(k, n + 1, k))

This post is licensed under CC BY 4.0 by the author.