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

문제

정수를 저장한 배열, arr에서 가장 작은 수를 제거한 배열을 리턴하는 함수, solution을 완성해주세요. 단, 리턴하는 배열이 빈 배열인 경우엔 배열에 -1을 채워 리턴하세요.

예를 들어 arr이 [4,3,2,1]인 경우는 [4,3,2]를 리턴하고, [10]면 [-1]을 리턴합니다.

 

제한사항

• arr은 길이 1 이상인 배열입니다.
• 인덱스 i, j에 대해 i != j이면 arr[i] != arr[j]입니다.

문제 링크

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

프로그래머스

 

풀이[c++]

#include <vector>

using namespace std;

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

내가 이 문제를 푼 방식은 무식한 방식이다.

 

arr의 모든 원소를 비교해 가장 작은 수를 찾아서 min에 넣고, arr의 모든 원소를 answer에 하나씩 넣는다.

만약, 그 원소가 가장 작은 수라면 다시 pop_back한다.

그 후, answer.size() = 0이면 push_back(-1)을 한다.