[C++][Programmers][스택/큐] 같은 숫자는 싫어

*문제

https://school.programmers.co.kr/learn/courses/30/lessons/12906

프로그래머스

 

*풀이방법

맨 처음에 arr 벡터에 있는 값을 answer 벡터에 넣고 arr 벡터를 반복문으로 돌면서 answer 벡터의 마지막 값과 다르면 또 넣도록 구현했다.

vector(stack)을 이용한 방법으로 vector의 push_back(), back(), front() 함수를 사용했다. 

 

* 알아둘 것

1) vector 함수들

size() - 벡터의 요소의 개수 반환 swap(vector객체) - 두 벡터의 내용 교환 empty() - 벡터가 비었는지 여부 반환 at(index) - index번째 요소 접근 front() - 벡터의 첫 번째 요소 반환 back() - 벡터의 마지막 요소 반환 begin() - 벡터의 첫 번째 요소 end() - 벡터의 마지막 요소

 

2) vector의 중복원소 제거 방법

나는 answer 벡터를 하나 더 두고 반복문을 돌면서 마지막 원소와 다른지를 판단했는데,

vector의 중복원소를 algorithm STL의 vector erase, unique 함수를 이용하면 중복제거가 된다더라.

 

원리를 알아보자.

ex) 1 1 2 3 3 4 5 5 5 6

 ->  1 2 3 4 5 6 5 5 5 6 (unique 적용)

vector의 unique 함수를 이용하면 중복 원소를 제거하며 앞부터 채워나간다.

그럼 원래 길이보다 중복된 원소의 개수만큼 줄어들게 되고 나머지 자리에는 기존 원소값들이 채워진다.

즉, 줄어든 개수를 알아내서 원래 길이에서 줄어든 개수만큼을 뺀 부분을 지우면 된다.

 

코드로 봐보자.

#include <vector>
#include <algorithm>

int main(){
    vector <int> v;
    
    v.push_back(1);
    v.push_back(1);
    v.push_back(2);
    v.push_back(3);
    v.push_back(3);
    v.push_back(4);
    v.push_back(5);
    v.push_back(5);
    v.push_back(5);
    v.push_back(6); // 1 1 2 3 3 4 5 5 5 6
   
    v.erase(unique(v.begin(), v.end()), v.end()); // 1 2 3 4 5 6
}

위처럼 unique 함수를 적용해서 중복된 원소개수만큼 길이가 줄어든 vector의 끝부분부터 마지막 값까지 지우도록 erase함수를 적용하면 된다.

 

* 코드

#include <vector>
using namespace std;

vector<int> solution(vector<int> arr) {
	vector<int> answer;

	if (!arr.empty()) answer.push_back(arr.front());
	for (auto& i : arr) {
		if (answer.back() != i) {
			answer.push_back(i);
		}
	}

	return answer;
}

중복제거를 반복문으로 구현한 코드

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

vector<int> solution(vector<int> arr) {
    arr.erase(unique(arr.begin(), arr.end()),arr.end());
    return arr;
}

erase, unique 함수로 중복제거한 코드