[c++]같은 숫자는 싫어

출처: 프로그래머스

 

문제 설명

배열 arr가 주어집니다. 배열 arr의 각 원소는 숫자 0부터 9까지로 이루어져 있습니다. 이때, 배열 arr에서 연속적으로 나타나는 숫자는 하나만 남기고 전부 제거하려고 합니다. 단, 제거된 후 남은 수들을 반환할 때는 배열 arr의 원소들의 순서를 유지해야 합니다. 예를 들면,

• arr = [1, 1, 3, 3, 0, 1, 1] 이면 [1, 3, 0, 1] 을 return 합니다.
• arr = [4, 4, 4, 3, 3] 이면 [4, 3] 을 return 합니다.

배열 arr에서 연속적으로 나타나는 숫자는 제거하고 남은 수들을 return 하는 solution 함수를 완성해 주세요.

제한사항

• 배열 arr의 크기 : 1,000,000 이하의 자연수
• 배열 arr의 원소의 크기 : 0보다 크거나 같고 9보다 작거나 같은 정수

 

입출력 예

arr	answer
[1,1,3,3,0,1,1]	[1,3,0,1]
[4,4,4,3,3]	[4,3]

입출력 예 설명

입출력 예 #1,2
문제의 예시와 같습니다.

 

문제풀이

#include <vector>
#include <iostream>
#include <queue>

using namespace std;

vector<int> solution(vector<int> arr) 
{
    vector<int> answer;
    queue<int> myQueue;
    
    for(int i = 0; i < arr.size(); i++)
    {
        if(myQueue.empty() || myQueue.back()!=arr[i]) //큐의 맨뒤 값과 배열의 값이 일치 하지 않으면 저장
        {
            myQueue.push(arr[i]);
        }
    }
     while (!myQueue.empty()) {
        answer.push_back(myQueue.front());  // 큐의 맨앞 요소를 배열 뒤로 추가
        myQueue.pop();  // 추가한 요소 제거
    }
    return answer;
}

이번 문제는 간단하면서, 스택/큐에 대해 한번 공부해볼만한 문제였다. 처음엔 스택으로 문제를 해결하려 했는데, 배열을 저장하면서 답이 역순으로 저장이 되었다. 그래서 값을 저장후 reverse를 해줘야하는데 그럴거면 큐를 이용해서 처음부터 저장을 정순으로 하고자 중간에 스택에서 큐로 바꿨다. 간단하면서 스택, 큐를 처음 공부하기 좋은 문제이지 않나 싶다.