문제
접근
배열을 순회하며 answer의 마지막 원소와 비교하며 넣을지 버릴지 정해주면 된다.
코드
- C++ 코드
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
#include <vector>
#include <iostream>
using namespace std;
vector<int> solution(vector<int> arr)
{
vector<int> answer;
int idx;
for(int i = 0; i<arr.size(); i++){
if(answer.empty()){
answer.push_back(arr[i]);
idx = 0;
continue;
}
if(answer[idx] != arr[i]){
answer.push_back(arr[i]);
idx++;
}
}
return answer;
}
- 파이썬 코드
1
2
3
4
5
6
7
8
9
def solution(arr):
answer = []
for item in arr:
if len(answer) == 0:
answer.append(item)
continue
if item != answer[-1]:
answer.append(item)
return answer