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

문제 설명

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

https://school.programmers.co.kr/learn/courses/30/lessons/12935

프로그래머스

제한 조건

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

입출력 예

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

 

내가 작성한 코드

import java.util.*;

class Solution {
    public ArrayList<Integer> solution(int[] arr) {
        ArrayList<Integer> answer = new ArrayList<Integer>();
        int min = arr[0];
        
        for (int num : arr) {
            if (num < min) {
                min = num;
            }
        }
        
        if (arr.length == 1 && arr[0] == 10) {
            answer.add(-1);
        }
        
        for (int i = 0; i < arr.length; i++) {
            if (arr[i] != min) {
                answer.add(arr[i]);
            }
        }
        return answer;
    }
}

코드 설명

• ArrayList<Integer> answer = new ArrayList<Integer>( ) : 동적리스트 ArrayList를 사용하여 answer를 생성한다.
• int min = arr[0] : 가장 작은 수를 구하기 위해 변수 min의 초기값으로 배열 arr의 0번째 인덱스 값으로 설정했다.
• for (int num : arr) { } : 반복문을 사용해서 배열 arr의 값들을 확인한다.
  • if (num < min) { min = num; } : 조건문을 사용해서 변수 min보다 num 값이 작다면 값을 교체해 준다.
• if (arr.length == 1 && arr[0] == 10) { answer.add(-1) } : 조건문을 사용해서 배열 arr의 길이가 1이고 0번째 인덱스 값이 10이라면 동적리스트 answer에 -1 값을 추가한다.
• for (int i = 0; i < arr.length; i++) { } : 반복문을 사용해서 변수 i가 0부터 arr.length - 1까지 반복한다.
  • if (arr[i] != min) { answer.add(arr[i]); } : 조건문을 사용해서 배열 arr의 i번째 값이 변수 min의 값과 같지 않을 경우에만 동적리스트 answer에 값을 추가한다.