(JAVA) 정수를 저장한 배열, arr 에서 가장 작은 수를 제거한 배열을 리턴하는 함수

▶ 문제

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

▶ 소스코드

class Solution {
    public int[] solution(int[] arr) {
        // 배열의 길이가 1인 경우 [-1]을 반환
        if (arr.length == 1) {
            return new int[]{-1};
        }

        // 배열에서 가장 작은 값을 찾음
        int min = arr[0];  // 배열의 첫 번째 값을 가장 작은 값으로 초기 설정
        for (int i = 1; i < arr.length; i++) {  // 배열을 순회하며 최소값을 찾음
            if (arr[i] < min) {
                min = arr[i];
            }
        }

        // 가장 작은 값이 제거된 새로운 배열을 생성
        int[] answer = new int[arr.length - 1];  // 새로운 배열의 크기는 기존 배열에서 하나 작음
        int idx = 0;  // 새로운 배열의 인덱스를 관리하기 위한 변수
        for (int i = 0; i < arr.length; i++) {
            if (arr[i] != min) {  // 배열에서 최소값이 아닌 값만 새로운 배열에 추가
                answer[idx++] = arr[i];  // 값을 넣고 인덱스 증가
            }
        }

        return answer;  // 새로운 배열 반환
    }
}

▶ 설명

1. 배열 길이가 1인 경우 처리 (if (arr.length == 1))
   • 먼저 배열이 하나의 원소만 가진 경우에는 그 원소를 제거할 수 없으므로 [-1]을 반환해야 함. 따라서 배열의 길이가 1인 경우를 먼저 확인하고, new int[]{-1}을 반환함.
   • 예를 들어, 배열이 [10]인 경우 [-1]을 반환함.
2. 배열에서 가장 작은 값 찾기
   • int min = arr[0];로 배열의 첫 번째 값을 최소값으로 설정한 후, for 루프를 통해 배열의 나머지 원소를 확인함.
     • 루프 안에서 if (arr[i] < min)을 사용해 현재 원소가 최소값보다 작은 경우, 그 값을 최소값으로 업데이트함.
     • 예시: arr이 [4, 3, 2, 1]일 때, 루프가 종료되면 min은 1이 됨.
3. 가장 작은 값이 제거된 배열 생성
   • 배열에서 가장 작은 값을 제외하고 나머지 값을 담기 위해 새로운 배열 answer를 만듭니다. 이 배열의 크기는 기존 배열보다 하나 작음 >>> new int[arr.length - 1].
   • 새로운 배열에 값을 넣을 때 사용할 인덱스 변수 idx를 0으로 초기화함.
   • 다시 for 루프를 돌면서 if (arr[i] != min) 조건에 따라, 최소값이 아닌 값만 새로운 배열에 복사.
   • 예시: [4, 3, 2, 1]에서 최소값 1을 제외하면 새로운 배열은 [4, 3, 2].
4. 최종 배열 반환
   • 모든 값이 처리된 후, 새로운 배열 answer를 반환