[프로그래머스] 133502 파이썬 - 햄버거 만들기

프로그래머스 133502

 

 

1은 빵, 2는 야채, 3은 고기

빵 - 야채 - 고기 - 빵 순서로 쌓인 햄버거만 포장한다.

 

풀이

1. hamburger라는 배열에 재료를 하나씩 넣는다.

2. 배열의 뒤에서부터 4개를 가져와 햄버거로 포장 가능한지 검사한다. (isBurger 함수)

3. 만약 포장 가능하다면 answer += 1, 포장된 재료 4개는 pop() 해준다.

 

코드

def solution(ingredient):
    answer = 0
    hamburger = []
    
    for i in ingredient:
        hamburger.append(i)
        if isBurger(hamburger[-4:]):
            answer += 1
            for _ in range(4):
                hamburger.pop()
    return answer

def isBurger(array):
    if array == [1,2,3,1]:
        return True
    else:
        return False

 

 

프로그래머스