[Programmers] 같은 숫자는 싫어 (c++)

https://school.programmers.co.kr/learn/courses/30/lessons/12906

프로그래머스

방법 1

#include <vector>
#include <iostream>

using namespace std;

vector<int> solution(vector<int> arr)
{
    vector<int> answer;

    for (int i = 0; i < arr.size(); i++)
    {
        if (answer.empty())
        {
            answer.emplace_back(arr[i]);
        }
        else {
            if (arr[i] != answer.back())
            {
                answer.emplace_back(arr[i]);
            }
        }
    }

    return answer;
}

방법 2 . unique를 통한 해결

#include <vector>
#include <iostream>
#include <algorithm>

using namespace std;

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

 

unique 함수는 아래 참고

https://cplusplus.com/reference/algorithm/unique/

https://cplusplus.com/reference/algorithm/unique/

 

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

int main() {
    solution({ 1,1,3,3,0,1,1 });
}

위의 코드를 디버깅 해보면

unique함수를 통해 1,1,3,3,0,1,1이 1,3,0,1,0,1,1로 변경된 것을 볼 수 있다.

 

방법1, 2 모두 동일한 결과가 나왔다.