프로그래머스, 같은 숫자는 싫어(C++)

프로그래머스

 

 

#include <vector>
#include <algorithm>
 
using namespace std;
 
vector<int> solution(vector<int> arr) 
{
    vector<int> answer;
    
    arr.erase(unique(arr.begin(), arr.end()), arr.end());
    answer = arr;
 
    return answer;
}

unique 함수를 이용하여 중복되는 것을 빼는데 중요한 것은 정렬해서는 안된다. 만약에 정렬을 한 뒤, unique 실행해버리면 1, 1, 3 ,3 ,0, 1, 1 의 예에서 0, 1, 3이 나올 것이다. 

unique 함수는 인접한 값들이 중복이면 하나만 남기고 제거한다.