[프로그래머스 / C++] 배열 만들기 4

문제

배열 만들기 4 : 문제 링크

문제 분석

• 정수 배열 arr가 주어진다. arr를 이용해 새로운 배열 stk를 만들려고 한다. 변수 i를 만들어 초기값을 0으로 설정한 후 i가 arr의 길이보다 작으면 다음 작업을 반복한다.

  • 만약 stk가 빈 배열이라면 arr[i]를 stk에 추가하고 i에 1을 더한다.
  • stk에 원소가 있고, stk의 마지막 원소가 arr[i]보다 작으면 arr[i]를 stk의 뒤에 추가하고 i에 1을 더한다.
  • stk에 원소가 있는데 stk의 마지막 원소가 arr[i]보다 크거나 같으면 stk의 마지막 원소를 stk에서 제거한다.

• 위 작업을 마친 후 만들어진 stk를 return
• 정수 배열 arr과 작업을 통해 만들어 질 정수 배열 stk와 정수형 변수 i를 0으로 초기화. while loop의 조건문을 i < arr.size()로 설정하여, i가 arr의 길이보다 작으면 반복하도록 설정. if ~ else if문과 empty() 함수를 사용하여 위와 같이 각 조건에 맞는 작업을 수행. loop 탈출 후, 최종적으로 저장된 stk를 return

empty() 함수 사용법
str.empty();
v.empty();

• string 또는 vector 객체가 비어있으면 1을, 문자열 또는 원소가 저장되어 있으면 0을 반환

풀이

#include <vector>

using namespace std;

vector<int> solution(vector<int> arr) {
    vector<int> stk;
    int i = 0;
    
    while(i < arr.size()) {
        if(stk.empty()) {
            stk.push_back(arr[i]);
            i++;
        }
        else if(stk.empty() == 0 && stk[stk.size() - 1] < arr[i]) {
            stk.push_back(arr[i]);
            i++;
        }
        else if(stk.empty() == 0 && stk[stk.size() - 1] >= arr[i]) {
            stk.pop_back();
        }
    }
    return stk;
}