코딩테스트/프로그래머스

[Lv.1] 제일 작은 수 제거하기

민톨이 2024. 8. 13. 01:54
728x90

📋 문제

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

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

 

📋 입출력 예시

📋 풀이1 (런타임 에러)

class Solution {
    public int[] solution(int[] arr) {
        // arr.length가 1 미만이면 false
        if(arr.length<=1){
            return new int[]{-1};
        }
        
        //최솟값 설정
        int min = arr[0];
        for(int i=0;i<arr.length;i++){
            if(min>arr[i]){
                min = arr[i];
            }
        }
        //정답 담을 answer 배열 선언
        int[] answer = new int[arr.length-1];
        for(int i=0;i<arr.length;i++){
            if(arr[i]!=min){
                answer[i] += arr[i];
            }
        }
        return answer;
    }
}

호기롭게 제출했으나 테스트케이스만 통과하고 런타임 에러 난 코드

칙쇼 ,, - !

 

answer[i] = arr[i]과정에서 배열의 인덱스를 초과하여 런타임오류가 발생

📋 풀이2

 public int[] solution(int[] arr) {
        // arr.length가 1 미만이면 false
        if(arr.length<=1){
            return new int[]{-1};
        }

        //최솟값 설정
        int min = arr[0];
        for(int i=0;i<arr.length;i++){
            if(min>arr[i]){
                min = arr[i];
            }
        }
        //정답 담을 answer 배열 선언
        int[] answer = new int[arr.length-1];
        //answer[i] = arr[i]대신에 idx값 ++식으로 순회
        //answer[idx++] = arr[i] 사용
        int idx = 0;
        for(int i=0;i<arr.length;i++){
            if(arr[i]!=min){
                answer[idx++] = arr[i];
            }
        }
        return answer;
    }

// 수정된 부분 설명

  1. 배열의 인덱스 관리:
    • answer[i] += arr[i]; 대신 answer[idx++] = arr[i];로 수정.
    • 이로써 answer 배열의 인덱스를 초과하지 않으면서 값을 정확히 대입할 수 있다
  2. 최솟값을 제외한 값만 추가:
    • arr[i] != min 조건을 만족하는 경우에만 answer 배열에 값을 추가하며, idx 변수를 사용해 추가할 위치를 추적한다