[프로그래머스] 배열 만들기 6(c++)

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

프로그래머스

---

---

답

#include <string>
#include <vector>

using namespace std;

vector<int> solution(vector<int> arr) {
    vector<int> stk;
    
    for(int i=0;i<arr.size();i++){
        if(stk.empty()==true){
            stk.push_back(arr[i]);
        }
        else{
            if(stk.back()==arr[i]){
                stk.pop_back();
            }
            else if(stk.back()!=arr[i]){
                stk.push_back(arr[i]);
            }
        }
    }

    if(stk.empty()==true){
        stk.push_back(-1);
    }
    
    return stk;       

}

 

vector stk를 만들어준다.

arr의 크기보다 작을떄 돌아가는 for문ㅇ르 돌린다.(i)

문제에 나와있던 조건에 맞는 if문ㅇ르 만들어주었다.

조건에 따른 작업을 수행했다.

​

마지막에 stk가 비어있으면 -1를 넣어주었다.

​

​

참고

쉬운문제였다.

그러나

배열이 비어있을떄 -1을 반환하는 부분에서 혼란스러울 수 이다.

if(stk.empty()==true){
    return -1;
}

이렇게 쓰면 오류가 났을 것이다.

​

왜냐하면 처음 함수를 만들데

vector형식을 반환하겠다고 하였기때문에

-1을 반환하려고 하면 오류가 나는 것이다.

​

그렇기 때문에 vector에 -1을 넣어주고 반환해주어야 한다.

---

끝