C++ / 프로그래머스 / 같은 숫자는 싫어

문제

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

코딩테스트 연습 - 같은 숫자는 싫어

문제풀이

연달아 같은 숫자를 출력하지만 않으면 중복된 숫자는 있을 수 있다.

먼저 첫 번째 숫자를 answer에 push 한 뒤, answer의 마지막 원소와 배열의 인덱스 값을 확인하면 된다.

만약 같으면 넘어가고, 다를 경우에만 answer에 push 한다.

 

코드

#include <vector>
#include <iostream>

using namespace std;

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

    answer.push_back(arr[0]);
    for (int idx = 0; idx < arr.size(); ++idx)
        if (answer.back() != arr[idx])
            answer.push_back(arr[idx]);

    return answer;
}