반갑습니다!

[프로그래머스] 가장 작은 수 제거하기 본문

알고리즘 문제 풀이

[프로그래머스] 가장 작은 수 제거하기

김덜덜이 2020. 4. 7. 16:25
프로그래머스
코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.
programmers.co.kr

풀이

min_element() 함수를 사용하면 iterator를 반환하므로 이를 이용하여 풀이하였다.

코드

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

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