[실력체크 level 1] 가장 작은 숫자 제거하기!

배열 arr가 주어졌을 때, 배열에서 가장 작은 숫자를 제거한 배열을 리턴하는 문제였다.

예를 들어, [4, -1, 2, 3]이 주어지면 -1을 제외하고 [4, 2, 3] 을 반환하면 된다.

단, [10] 과 같이 원소가 하나만 있는 배열이 주어져서 제거한 후 원소가 아무것도 없으면, 배열에 -1을 넣어서 반환해준다.

 

#include <string>
#include <vector>
#include <algorithm>
#include <climits>
 
using namespace std;
 
vector<int> solution(vector<int> arr) {
    vector<int> answer;
 
    int num = INT_MAX;
    int idx = 0;
    for(int i=0; i<arr.size(); i++){
        if(arr[i]<num){
            num = arr[i];
            idx = i;
        }
    }    
 
    arr.erase(arr.begin()+idx);
 
    if(arr.size()==0){
        arr.push_back(-1);
    }
 
    for(int i=0; i<arr.size(); i++){
        answer.push_back(arr[i]);
    }
 
    return answer;
}