뜨균이의 개발 생활

프로그래머스의 같은 숫자는 싫어 라는 문제를 풀어봤습니다.

링크

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

문제

풀이

같은 숫자가 연속해서 나타나면 한 번만 저장합니다.
같은 숫자 체크는 Stack을 사용하였고
숫자 저장은 Queue를 사용했습니다.

Stack에 숫자를 저장하고 pop(담긴 숫자 반환)으로 같은 숫자인지 확인합니다.
이 때 Stack에 아무것도 담겨있지 않은 상태에서 pop을 사용할 수 없으니 Stack에 값이 없는 경우는 별도로 처리하였습니다.
Stack에 담겨있지 않은 숫자만 Queue에 저장하여 반환하였습니다.

public int[] solution(int []arr) {
    Stack<Integer> stack = new Stack<>();
    Queue<Integer> queue = new LinkedList<>();

    for (int a : arr) {
        if (stack.empty()) {
            stack.add(a);
            queue.add(a);
            continue;
        }
        if (stack.pop() == a) {
            stack.add(a);
            continue;
        }
        stack.add(a);
        queue.add(a);
    }
    return queue.stream().mapToInt(m -> m).toArray();
}

후기

* 난이도 (5점 만점)

5 : 풀 줄 알면 기업 코딩 테스트는 문제없음.

4 : 평균적인 기업 코딩 테스트의 중간 이상.

3 : 평균적인 기업 코딩테스트의 쉬운 문제.

2 : 알고리즘 문제를 연습하고 있다면 풀 수 있는 문제.

1 : 시간이 오래 걸리지 않고, 누구나 풀 수 있는 문제.

프로그래머스의 "같은 숫자는 싫어"는 stack, queue 유형의 문제로 분류되어 있어서 stack과 queue를 사용하여 풀었습니다.