LV1 - 같은 숫자는 싫어

배열에서 반복되는 수를 제거하여 나열하라. 
algorithm 라이브러리에서 'unique'라는 키워드를 이용해 배열을 중복된 수가 없이 나열시킨다. 그 후  erase라는 vector 명령어를 통해 나열되고 필요없는 데이터들은 지운다. 굉장히 깔끔하고 멋진 코드인 것 같다.
 
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
 
vector<int> solution(vector<int> arr);
 
int main()
{
    vector<int> arr = { 1,1,3,3,0,1,1 };
    solution(arr);
 
return 0;
}
 
vector<int> solution(vector<int> arr)
{
    arr.erase(unique(arr.begin(), arr.end()), arr.end());
    vector<int> answer = arr;
    return answer;
}

댓글을 작성해보세요.