[프로그래머스 코딩테스트] 같은 숫자는 싫어(Java)

  문제 설명

  배열 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보다 작거나 같은 정수

 

  입출력 예

arr	answer
[1, 1, 3, 3, 0, 1, 1]	[1, 3, 0, 1]
[4, 4, 4, 3, 3]	[4, 3]

import java.util.*;

public class Solution {
    public int[] solution(int []arr) {
        Stack<Integer> stack = new Stack<>();
        int count = 0;
        
        for (int i = 0; i < arr.length; i++) {
            if (stack.isEmpty() || stack.peek() != arr[i]) {
                stack.push(arr[i]);
                count++;
            }
        }
        int[] answer = new int[count];
        int index = count - 1;
        for (int i = index; i >= 0; i--) {
            answer[i] = stack.pop();
        }
        return answer;
    }
}

  이 문제는 스택을 이용해 풀었습니다. 큐로 풀어보려고도 했는데 잘 안되더라구요.

  우선 answer 배열의 크기가 될 count 변수를 생성해줍니다. for문은 arr의 길이만큼 순회합니다. 만약 스택이 비어있거나, 스택의 마지막 값과 arr[i]의 값을 비교하여 같지 않다면(연속적인 숫자가 삽입되지 않도록) arr[i]를 stack에 삽입합니다. count가 1씩 증가합니다.

  위에서 증가시킨 count를 answer 배열의 크기로 할당합니다. stack은 후입선출(Last In First Out)의 구조입니다. 문제에서는 배열 원소들의 순서를 유지하고 싶어하므로, answer 배열에 할당할 때, 마지막 인덱스부터 역순으로 진행합니다. 따라서 index 변수를 생성한 뒤 count - 1로 초기화하여 i에 할당합니다. i가 0보다 크거나 같아질 때까지 순회하며 i의 값이 1씩 감소합니다. stack에서 pop 메소드로 꺼낸 원소가 answer의 i번째 인덱스에 할당됩니다.