[Programmers - lv01] 같은 숫자는 싫어 (cpp / python)

같은 숫자는 싫어

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

문제 설명

.

문제 풀이

소스코드 : C++

#include <vector>
#include <iostream>

using namespace std;

vector<int> solution(vector<int> arr) {
    vector<int> answer;
    answer.push_back(arr[0]);
    for(int i=1; i<arr.size(); i++){
        if(arr[i-1] !=arr[i])answer.push_back(arr[i]);
    }
    return answer;
}

소스코드 : Python

간단한 문제지만 파이썬으로 한 줄 코딩을 할 방법을 찾다가 조금 억지?로 끼워맞췄다.

def solution(arr):
    return [arr[0]] + [arr[i] for i in range(1,len(arr)) if arr[i-1] != arr[i]]