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

문제

배열 만들기 6 : 문제 링크

문제 분석

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

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

• 위 작업을 마친 후 만들어진 stk를 return (단, 만약 빈 배열을 return 해야한다면 [-1]을 return
• 정수 배열 arr과 작업을 통해 만들어 질 정수 배열 stk와 정수형 변수 i를 0으로 초기화. while loop의 조건문을 i < arr.size()로 설정하여, i가 arr의 길이보다 작으면 loop를 반복하도록 설정. if ~ else if문과 empty() 함수를 사용하여 위와 같이 각 조건에 맞는 작업을 수행. loop 탈출 후, skt가 빈 배열이라면 -1을 저장한 후, 최종적으로 저장된 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.pop_back();
            i++;
        }
        else if(stk.empty() == 0 && stk[stk.size() - 1] != arr[i]) {
            stk.push_back(arr[i]);
            i++;
        }
    }
    if(stk.empty()) stk.push_back(-1);
    return stk;
}