[Programmers/Java/Lv.1/문자열] 61.햄버거 만들기

문제 설명

프로그래머스

 

시간 초과 나의 풀이

import java.util.*;

class Solution {
    public int solution(int[] ingredient) {
        int answer = 0;
        List<Integer> list = new ArrayList<>();
        
        for (int ing : ingredient) {
            list.add(ing);
        }
        
        while (true) {
            boolean isValid = false;
            for (int i = 0; i < list.size() - 3; i++) {
                if (list.get(i) == 1 && list.get(i+1) == 2 && list.get(i+2) == 3 && list.get(i+3) == 1) {
                    // 순서대로 삭제하면 인덱스 밀리므로 같은 위치 4번 삭제
                    list.remove(i); // 1
                    list.remove(i); // 2
                    list.remove(i); // 3
                    list.remove(i); // 1
                    answer++;
                    isValid = true;
                    break; // for문 벗어남, 햄버거 만들었으면 다시 처음부터
                }
            }
            if (!isValid) break; // while문 벗어남
        }
        
        return answer;
    }
}

 

풀이법

import java.util.*;

class Solution {
    public int solution(int[] ingredient) {
        int answer = 0;
        Stack<Integer> stack = new Stack<>();
        
        for (int ing : ingredient) {
            stack.push(ing);
            
            // 스택에 4개 이상 쌓였을 때만 검사
            if (stack.size() >= 4) {
                int size = stack.size();
                if (stack.get(size - 4) == 1 &&
                    stack.get(size - 3) == 2 &&
                    stack.get(size - 2) == 3 &&
                    stack.get(size - 1) == 1) {
                    
                    // 햄버거 재료 제거
                    stack.pop();
                    stack.pop();
                    stack.pop();
                    stack.pop();
                    
                    answer++;
                }
            }
        }
        
        return answer;
    }
}