햄버거 만들기-파이썬(Python) [프로그래머스/코딩테스트 연습/Lv.1]

문제

https://school.programmers.co.kr/learn/courses/30/lessons/133502?language=python3

프로그래머스

 

---

풀이

def solution(ingredient):
    answer=0
    li=[]
    for i in ingredient: # 재료를 앞에서부터 하나씩 확인
        li.append(i)
        # li의 마지막 네 개의 원소들을 확인하여 각각 1,2,3,1 이면 그것들을 빼주고 정답을 1 더해준다.
        if len(li)>=4 and li[-4]==1 and li[-3]==2 and li[-2]==3 and li[-1]==1: 
            for _ in range(4):
                li.pop()
            answer+=1
    return answer

---

Tip

ingredient의 원소들을 확인하면서 연속으로 된 1, 2, 3 ,1이 있을 경우 그것을 제거하고 정답을 더해주는 식으로 문제를 풀면 원소들을 제거하는 함수의 시간복잡도 때문에 시간 초과가 발생한다. 따라서 다른 리스트(li)를 만들어서 ingredient의 원소를 하나씩 li에 넣어가면서 li의 마지막 원소 네 개가 1, 2, 3, 1인 경우에 pop() 함수를 이용하여 원소들을 제거해주면 시간 초과가 발생하지 않고 문제를 풀 수 있다.

---

TIL

없음