[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.