문제풀이/프로그래머스

[프로그래머스] 같은 숫자는 싫어

Hyeon-Uk 2020. 11. 24. 01:08
반응형

 

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

 

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

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

programmers.co.kr


문제풀이)

이 문제는 c++에서 주어지는 메소드를 얼마나 잘 활용하나 문제인것 같다.

 

unique(st,end) => [st,end) 범위의 연속된 값을 뒤로 빼준뒤, 쓰레기값들의 위치를 반환해준다.

ex) 1,1,3,3,0,1,1 -> 1,3,0,1,1,1 // 4번째 인덱스 반환

※algorithm헤더파일을 include해야된다.※

 

erase(st,end) => [st,end) 범위를 제거해준다.

 

이 메소드들을 잘 활용해서 연속된 값을 제거해주면 된다.

 

#include <vector>
#include <iostream>
#include<algorithm>
using namespace std;

vector<int> solution(vector<int> arr) 
{
    vector<int> answer;
    
    arr.erase(unique(arr.begin(),arr.end()),arr.end());
    answer=arr;
    return answer;
}
반응형