Programers : 제일 작은 수 제거하기

  • <algorithm>의 find는 찾으면 해당 요소의 iterator를 반환 /
    찾지 못하면 v.end()를 가리키게 된다!
  • <algorithm>에 해당 stl에 min/max를 찾는 함수가 있다!
    1) min_element() : 최소값에 해당하는 요소의 iterator를 반환
    2) max_element() : 최대값에 해당하는 요소의 iterator를 반환

코드

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

vector<int> solution(vector<int> arr) {
    vector<int> answer;
    if(arr.size() > 1)
    {
        int min_value = *min_element(arr.begin(), arr.end());
        arr.erase(find(arr.begin(), arr.end(), min_value));
        answer = arr;
    }
    else
        answer.push_back(-1);
    return answer;
}

좋은 웹페이지 즐겨찾기