[프로그래머스 Lv.1 / C++] Day37. 제일 작은 수 제거하기

2024. 07. 10 - 코딩테스트 스터디 Day37

 
 

01. 문제 설명

정수를 저장한 배열, arr 에서 가장 작은 수를 제거한 배열을 리턴하는 함수, solution을 완성해 주세요. 단, 리턴하려는 배열이 빈 배열인 경우엔 배열에 -1을 채워 리턴하세요. 예를 들어 arr이 [4,3,2,1]인 경우는 [4,3,2]를 리턴하고, [10]면 [-1]을 리턴합니다.

 

 

02. 제한사항

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

 

 

03. 입출력 예

arr	return
[4, 3, 2, 1]	[4, 3, 2]
[10]	[-1]

 

 

04. 풀이 및 답

처음에는 arr를 정렬해서 빼내야 하나 싶었는데, 문제에서 리턴되는 배열의 순서가 딱히 명시되어 있지 않기 때문에 다른 방법으로 풀어봤다.

 

#include <string>
#include <vector>
#include <algorithm>

using namespace std;

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

min_element 함수를 사용하기 위해 <algorithm> 라이브러리 헤더파일을 포함했다.

 

vector의 erase 함수를 통하여 벡터 속 가장 작은 값을 제거해 준다.

(가장 작은 값을 찾기 위해서는 min_element, 가장 큰 값을 찾기 위해서는 max_element를 사용하면 된다)

 

제거 후, 벡터에 남은 값이 없다면 push_back 함수를 이용하여 -1을 벡터에 삽입해 준다.

 

 

야호

 

---

끝!