[레벨1] 같은 숫자는 싫어

 

 

#include <vector>
#include <iostream>

using namespace std;
vector<int> solution(vector<int> arr)
{
    vector<int> answer;
    int n = arr.size();
    int i, pos = 0;
    for (i = 0; i < n - 1; ++i)
    {
        if (arr[i] != arr[i + 1])
        {
            answer.push_back(arr[i]);
            pos = i + 1;
        }
    }
    int index = answer.size() - 1;
    if (arr[pos] != answer[index] || index == -1)
        answer.push_back(arr[pos]);
    return answer;
}

 

 

다른 사람의 풀이에서 매우 좋은 함수를 보았는데. 바로 unique함수.

    arr.erase(unique(arr.begin(), arr.end()),arr.end());

    vector<int> answer = arr;
    return answer;