[프로그래머스] LV1 햄버거 만들기

문제

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

프로그래머스

 

접근

- 처음에는 ingredient에 대해서 for문 돌면서 1이 나오면 그때 index를 체크하고 그거 부터 뒤로 3개를 체크하면서 검사를 하려고 했는데, 구현하기에 너무 복잡했다

- 구글링을  해버렸다... 무튼 스택의 기능을 쓰기에 너무 좋은 문제였다...stack을 떠올리지 못한 나를 반성하자

 

풀이

#1231 되면 ans += 1
def solution(ingredient):
    ans = 0
    
    stack = []
    for i in ingredient:
        stack.append(i)
        if stack[-4:] == [1,2,3,1]:
            ans += 1
            for i in range(4):
                stack.pop()
    
    return ans