[프로그래머스] 같은 숫자는 싫어

jh Seo·2023년 6월 16일
0

프로그래머스

목록 보기
8/33
post-custom-banner

개요

프로그래머스: 같은 숫자는 싫어

  • 문제 설명
    배열 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 함수를 완성해 주세요.

접근방식

  1. 딱 떠오른 건 set이였다.
    하지만 set은 저장할때 key값으로 정렬이 되어버리므로 사용할 수 없었다.

  2. 따라서 큐를 사용하여 풀었다.
    arr벡터의 모든값을 큐에 저장 후, 큐의 front원소를 저장하고,
    큐의 front원소와 같으면 전부 pop하면서 진행한다.

       while(!q.empty()){
           int curTop=q.front();
           while(!q.empty()&&curTop==q.front()){
               q.pop();
           }
           answer.push_back(curTop);
       }

    꼭 curTop이 q.front()와 같은지 체크할때 큐가 비어있는지 체크를 같이 해야한다.
    비어있는데 pop하면 segmentation fault오류가 뜨므로 조심해야한다.

전체코드

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

using namespace std;

vector<int> solution(vector<int> arr) 
{
    vector<int> answer;
    queue<int> q;
    for(auto iter= arr.begin();iter!=arr.end();++iter){
        q.push(*iter);
    }
    while(!q.empty()){
        int curTop=q.front();
        while(!q.empty()&&curTop==q.front()){
            q.pop();
        }
        answer.push_back(curTop);
    }
    
    return answer;
}

문풀후생

다른 방법으로는 예전에 정리했떤 unique함수와 erase함수를 이용하여
짧게 풀 수 있다.

마침 모든 중복값도 아니고 연속된 중복값만 제거하므로 erase와 unique함수를 조합해

vector<int> solution(vector<int> arr) 
{
    vector<int> answer;
    arr.erase(unique(arr.begin(),arr.end()),arr.end());
    answer=arr;
    return answer;
}

이런식으로 풀이가 가능하다.

profile
코딩 창고!
post-custom-banner

1개의 댓글

comment-user-thumbnail
2023년 9월 21일

재현이 화이팅~
사룽~ ㅎ
나두공부하러가야지~
걍다때려치고 너랑 애버랜드가고싶긴하네 ㅜ ㅋㅋ

답글 달기