[C/C++] programmers Level 1 같은 숫자는 싫어

 

 

 

코드

 

#include <vector>
#include <stack> // 스택

using namespace std;

vector<int> solution(vector<int> arr) 
{
    vector<int> answer;
    stack<int> check;
    
    check.push(arr[0]);
    answer.push_back(arr[0]);
    
    for(int i=1;i<arr.size();i++){
        if(check.top()!=arr[i]){
            check.push(arr[i]);
            answer.push_back(arr[i]);
        }
    }
    
    return answer;
}