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

김주란·2021년 11월 11일
0

Algorithm

목록 보기
9/13

문제 설명

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

제한 조건

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

입출력 예

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

풀이

  1. Math.min() 이용 - 답은 나오나 테스트 케이스 실패...
class Solution {
    public int[] solution(int[] arr) {
        if(arr.length == 1)
            return new int[] {-1};

        int[] result = new int[arr.length-1];
        int min = arr[0];
        for(int i = 0; i < arr.length; i++) {
            min = Math.min(arr[i], min);
        }
        
        int idx = 0;
        for(int i= 0;i< result.length; i++) {
            if(arr[i] != min) {
                result[idx] = arr[i];
                idx++;
            }
        }
        return result;
    }
}
  1. 첫번째 원소를 제일 작은 값으로 가정하고 시작하기
import java.util.*;

class Solution {
    public int[] solution(int[] arr) {
               if(arr.length == 1) {
            int[] result = {-1};
            return result;
        }
        ArrayList<Integer> list = new ArrayList<>();
        for(int i = 0; i <arr.length; i++) {
            list.add(arr[i]);
        }
        //첫 번째 값을 제일 작은 값으로 가정.

        int min = 0;
        for(int i = 1; i <arr.length; i++) {
            if(arr[min] > arr[i]) {
     // 양수라는 것은 왼쪽 값이 더 크다는 얘기니깐 min값을 더 작은 i 값으로 바꿔줌.
                min = i;
            }
        }
        list.remove(min); //
        System.out.println(list.size());
        int[] result = new int[list.size()];
        for(int i = 0; i < list.size(); i++) {
            result[i] = list.get(i);
        }
        return result;
    }
}
  1. stream과 lamda 사용
import java.util.Arrays;
        import java.util.stream.Stream;
        import java.util.List;
        import java.util.ArrayList;

class Solution {
    public int[] solution(int[] arr) {
        if (arr.length <= 1) return new int[]{ -1 };
        int min = Arrays.stream(arr).min().getAsInt();
        return Arrays.stream(arr).filter(i -> i != min).toArray();
    }
}

0개의 댓글