문제 설명
배열 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 함수를 완성해 주세요.
딱 떠오른 건 set이였다.
하지만 set은 저장할때 key값으로 정렬이 되어버리므로 사용할 수 없었다.
따라서 큐를 사용하여 풀었다.
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;
}
이런식으로 풀이가 가능하다.
재현이 화이팅~
사룽~ ㅎ
나두공부하러가야지~
걍다때려치고 너랑 애버랜드가고싶긴하네 ㅜ ㅋㅋ