알고리즘/프로그래머스 1단계

프로그래머스 - 제일 작은 수 지우기 - C++

게임만드는학생 2023. 10. 4. 21:25

https://school.programmers.co.kr/learn/courses/30/lessons/12935

 

프로그래머스

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

programmers.co.kr

 

#include <string>
#include <vector>

using namespace std;

vector<int> solution(vector<int> arr) {
    vector<int> answer;
    int min=2100000000;
    int minId;
    for(int i=0;i<arr.size();i++)
    {
        if(min>arr[i])
        {
            min=arr[i];
            minId=i;
        }
    }
    
    arr.erase(arr.begin()+minId);
    if(arr.size()==0)
        arr.push_back(-1);
    return arr;
}

 

설명 

주어진 배열 중에서 가장 작은 수를 지우고 그 배열을 반환하는 문제이다. 

 

먼저 가장 작은 수의 위치를 알아낸 후에 vector.erase 함수를 이용해 그 원소를 제거한다.