C++(코딩테스트) - 최솟값 빼고 출력

내 풀이

#include <string>
#include <vector>

using namespace std;

vector<int> solution(vector<int> arr) {
    vector<int> answer;
    int min;

    if (arr.size() > 0) min = arr[0];

    for (int i = 0; i < arr.size(); i++)
    {
        if (min > arr[i]) min = arr[i];
    }

    for (int i = 0; i < arr.size(); i++)
    {
        if (arr[i] == min) arr.erase(arr.begin() + i);
    }

    if (arr.size() != 0)
    {
        answer = arr;
    }
    else
    {
        answer.push_back(-1);
    }
        return answer;
}

 

 

다른사람 풀이 ㄷㄷ

#include <algorithm>
#include <vector>
using namespace std;

vector<int> solution(vector<int> arr) {
    if (arr.size() == 1) {
        return { -1 };
    }
    arr.erase(min_element(arr.begin(), arr.end()));
    return arr;
}