Programmers / Level 1 / 제일 작은 수 제거하기

[문제설명]

정수를 저장한 배열, arr 에서 가장 작은 수를 제거한 배열을 리턴하는 함수, solution을 완성해주세요. 단, 리턴하려는 배열이 빈 배열인 경우엔 배열에 -1을 채워 리턴하세요. 예를들어 arr이 [4,3,2,1]인 경우는 [4,3,2]를 리턴 하고, [10]면 [-1]을 리턴 합니다.

[제한사항]

• arr은 길이 1 이상인 배열입니다.
• 인덱스 i, j에 대해 i ≠ j이면 arr[i] ≠ arr[j] 입니다.

[입출력 예]

arr                                                                                                      return

[4,3,2,1]	[4,3,2]
[10]	[-1]

import java.util.ArrayList;

class Solution {
    public int[] solution(int[] arr) {
        int min = arr[0]; 
        // 가장 작은 수 구하기
        for(int i=0; i<arr.length; i++) {
            if(arr[i] < min) {
                min = arr[i];
            }
        }
        
        ArrayList<Integer> list = new ArrayList<Integer>();
        for(int i=0; i<arr.length; i++) {
            if(arr[i] != min) { // 최솟값 제외하고
                list.add(arr[i]);
            }
        }
        
        int [] answer;
        if(list.size() == 0) {
            answer = new int[1];
            answer[0] = -1;
        }
        else {
            answer = new int[list.size()];
            for(int i=0; i<answer.length; i++) {
                answer[i] = list.get(i);
            }
        }
        
        return answer;
    }
}

빈 배열인 경우에 배열에 -1을 채워 리턴해야 하는데 그 부분에서 조금 헤맸다.

빈 배열인 경우와 그렇지 않은 경우를 나눠 answer 배열 생성을 달리 해줘야 한다.

 

ArrayList 사용법을 능숙하게 익히면 문제 푸는 데 도움이 많이 될 것 같다.