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

문제 링크 : https://programmers.co.kr/learn/courses/30/lessons/12906

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

 

풀이 방법

처음으로 시작하는 원소를 cont 변수에 넣고 해당 원소가 연속될 때까지 continue 시킨다.

원소가 달라지면 해당 원소를 answer 벡터에 넣고 cont 값을 해당 값으로 바꿔준다.

arr 원소의 크기는 0부터 9까지이므로 초기 cont값은 10으로 두면 된다.

 

소스 코드

#include <vector>
#include <iostream>

using namespace std;

vector<int> solution(vector<int> arr) 
{
    vector<int> answer;
    int cont = 10;
    
    for(int i=0; i<arr.size(); i++){
        if(arr[i] == cont){
            continue;
        }
        else{
            answer.push_back(arr[i]);
            cont = arr[i];
        }
    }

    return answer;
}