[Python] 햄버거 만들기

✏️ 문제

 

문제 파악

문제를 읽자마자 스택을 사용하는 코드라고 짐작했는데 ....

스택을 여러 개 사용해서 푸는 줄 알고 무슨 조건일 때 스택을 추가하며 .. 이런 고민 때문에 난이도에 비해 시간이 좀 걸렸다.

(스택 하나로도 풀 수 있는 문제였음 ..! 🥲)

 

 

코드

def solution(ingredient):
    answer = 0  
    stack = []

    for i in ingredient:
        if i != 1 and len(stack) == 0:
            continue
        else: 
            stack.append(i)

            if stack[-4:] == [1, 2, 3, 1]:
                answer += 1 
                for _ in range(4):
                    stack.pop()

    return answer