[프로그래머스] 햄버거 만들기(java)

문제

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

프로그래머스

---

나의 풀이

import java.util.Stack;

class Solution {
    public int solution(int[] ingredient) {
        int answer = 0;
        Stack<Integer> stack = new Stack<>();
        for (int in : ingredient) {
            stack.push(in);
            int size = stack.size();
            if (size > 3 && stack.peek() == 1 && stack.get(size - 2) == 3 && stack.get(size - 3) == 2
                    && stack.get(size - 4) == 1) {
                answer++;
                stack.pop();
                stack.pop();
                stack.pop();
                stack.pop();

            }
        }
        return answer;
    }
}

Stack 객체를 생성해서 들어오는 순서대로 담는다. 이 때 stack에 쌓인 크기가 3이 넘고 마지막에 들어온 데이터부터 순서가 1-3-2-1 (햄버거가 1-2-3-1로 쌓여야 하니까)이 되면 stack에서 4개를 모두 제거한다.

 

전달받은 배열 ingredient의 길이 n만큼 순회하므로 시간 복잡도는 O(n)이고, 선형이다.