728x90

같은 숫자는 싫어

 

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

배열 arr가 주어집니다. 배열 arr의 각 원소는 숫자 0부터 9까지로 이루어져 있습니다. 이때, 배열 arr에서 연속적으로 나타나는 숫자는 하나만 남기고 전부 제거하려고 합니다. 단, 제거된 후 남은

programmers.co.kr

문제 설명

.

문제 풀이

소스코드 : 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]]
728x90
300x250
WONILLISM