프로그래머스 - 짝수는 싫어요

 

---

 

import java.util.ArrayList;

class Solution {
    public static ArrayList<Integer> solution(int n) {
        ArrayList<Integer> answer = new ArrayList<>();
        for(int i = 1; i <= n; i++){
            if(i % 2 != 0){
                answer.add(i);
            }
        }
        
        return answer;
    }
}

 

ArrayList를 이용해 쉽게 짝수가 아닌 홀수만 추가할 수 있도록 해서 쉽게 풀 수 있었다.

 

만약 반환 타입을 ArrayList<Integer> 가 아닌 int[]를 이용하고싶다면 

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

class Solution {
    public int[] solution(int n) {
        List<Integer> answer = new ArrayList<>();
        for(int i=1; i<=n; i++){
            if(i % 2 == 1){
                answer.add(i);
            }
        }
        return answer.stream().mapToInt(x -> x).toArray();
    }
}

이렇게 return에 stream을 사용해서 리턴해주면 된다