Post

[Programmers] #120813 - 짝수는 싫어요 [Java][C++][Python]

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

[Programmers] #120813 - 짝수는 싫어요 [Java][C++][Python]

문제 링크


1. 아이디어

정수 n이 주어질 때, n 이하의 홀수를 오름차순으로 모아 배열로 반환하면 되는 문제다. 1부터 시작해 2씩 증가시키며 n을 넘지 않을 때까지 순회하면 조건을 만족하는 값만 이미 오름차순으로 얻어진다.


2. 복잡도

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

($N$ = 입력값 n)


3. 코드

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

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

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

        return arr;
    }
}

홀수의 개수가 항상 (n + 1) / 2개임을 이용해 배열 크기를 미리 정해뒀다.

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

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

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

range로 홀수만 뽑아 곧바로 리스트로 반환했다.


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