[백준 222번] (알고리즘)

Basic

문제

문제링크

문제 설명

정수 n이 매개변수로 주어질 때, n 이하의 홀수가 오름차순으로 담긴 배열을 return하도록 solution 함수를 완성해주세요.

---

제한사항

• 1 ≤ n ≤ 100

---

입출력 예

n	result
10	[1, 3, 5, 7, 9]
15	[1, 3, 5, 7, 9, 11, 13, 15]

---

입출력 예 설명

입출력 #1

• 10 이하의 홀수가 담긴 배열 [1, 3, 5, 7, 9]를 return합니다.

입출력 #1

• 15 이하의 홀수가 담긴 배열 [1, 3, 5, 7, 9, 11, 13, 15]를 return합니다.

접근법

우선 배열의 사이즈를 관리하기 쉽게 동적으로 만들어준다.
그 후 모듈러를 사용해 홀수만 담고 리턴하면 된다고 생각 했는데 리턴 타입이 배열로 고정되어 있어서 arrayList를 다시 배열로 변환시켜 준다..

구현 코드

import java.util.ArrayList;
import java.util.List;

class Solution {
    public int[] solution(int n) {
        List<Integer> answerList = new ArrayList<>();

        for (int item = 1; item <= n; item++) {
            if (item % 2 != 0) {
                answerList.add(item);
            }
        }

        // List를 배열로 변환
        int[] answer = new int[answerList.size()];
        for (int i = 0; i < answerList.size(); i++) {
            answer[i] = answerList.get(i);
        }

        return answer;
    }
}

def solution(n):
    return [x for x in range(n+1) if x % 2 != 0]

파이썬 까먹을까봐 파이썬도 같이 풀었다.

시간 복잡도와 공간 복잡도

• 시간 복잡도는 반복문 중첩이 없어서 O(n)
• 공간 복잡도는 answerList -> O(n) / answer배열 -> O(n) = O(n)

회고 과정

자바르 사용하는게 익숙하지 않아서 다른 사람들이 풀어 놓은 코드를 봤다.

import java.util.stream.IntStream;

class Solution {
    public int[] solution(int n) {
        return IntStream.rangeClosed(0, n).filter(value -> value % 2 == 1).toArray();
    }
}

스트림이 익숙해지 때까지 연습해야겠다.