알고리즘(Algorithm)/프로그래머스(Programmers)

[프로그래머스] 제일 작은 수 제거하기 - JAVA

나아가는중 2021. 11. 2. 16:08
반응형

프로그래머스 제일 작은 수 제거하기 - JAVA


문제 설명

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

제한 조건

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

문제 풀이

  1. 리턴하려는 배열이 빈 배열인 경우는 arr의 길이가 1인 경우로 검사 후 배열에 -1을 채워 리턴.
  2. arr배열에서 가장 작은 수를 찾아 저장한다.
  3. 반환하는 새로운 배열은 arr배열에서 가장 작은 수를 제거한 사이즈가 -1인 배열이다.
  4. 가장 작은 수를 제외한 모든 수를 answer배열에 저장 후 리턴한다.

소스 코드

class Solution {
    public int[] solution(int[] arr) {
        if (arr.length == 1) { // 1
            return new int[]{-1};
        } else {
            int min = Integer.MAX_VALUE; // 2
            for (int n : arr) {
                min = Math.min(n, min);
            }

            int[] answer = new int[arr.length - 1]; // 3

            int index = 0; // 4
            for (int n : arr) {
                if (n != min) {
                    answer[index++] = n;
                }
            }

            return answer;
        }
    }
}
반응형