주뇽's 저장소

프로그래머스 스쿨 고득점 스택/큐 같은 숫자는 싫어 본문

알고리즘/스택, 큐

프로그래머스 스쿨 고득점 스택/큐 같은 숫자는 싫어

뎁쭌 2023. 7. 4. 18:17
728x90
반응형

같은 숫자는 싫어

문제 설명

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

  • arr = [1, 1, 3, 3, 0, 1, 1] 이면 [1, 3, 0, 1] 을 return 합니다.
  • arr = [4, 4, 4, 3, 3] 이면 [4, 3] 을 return 합니다.

배열 arr에서 연속적으로 나타나는 숫자는 제거하고 남은 수들을 return 하는 solution 함수를 완성해 주세요.

제한사항

  • 배열 arr의 크기 : 1,000,000 이하의 자연수
  • 배열 arr의 원소의 크기 : 0보다 크거나 같고 9보다 작거나 같은 정수

같은 숫자는 싫어 문제는 스택을 이용하여 연속된 숫자는 제외하고 배열에 넣어 주기만 하면 되는 문제이다. 다만 주의해야 할 점은 중복된 숫자를 제외하는게 아닌 연속된 숫자를 제외하는 것이다. 예를 들어 1,1,3,1,1 의 경우 정답이 1,3이 아닌 1,3,1이다. 이 부분만 주의한다면 어렵지 않게 해결할 수 있다.

1. 현재 들어온 숫자가 연속되었는지 확인하기 위한 check 변수 선언

2. 반복문을 순회하면서 배열의 현재 값이 이전에 들어왔는지 확인

3. 만약 최초라면 스택이 push 하고 check 값 갱신

4. 최초가 아니라면 무시

#include
#include

using namespace std;
typedef long long ll;

vector solution(vector arr)
{
vector answer;
ll check = -1 ;
for(ll i=0; i<arr.size(); i++){
if(check != arr[i]) // 이전 숫자와 연속되는지 확인
answer.push_back(arr[i]); // 연속 되지 않았다면 추가
check = arr[i]; // check 변수 초기화
}
return answer;
}