알고리즘 (JAVA)/프로그래머스 알고리즘

[코테준비 - python] 햄버거만들기 level1 - 개발자 배찌

개발자 배찌 2022. 12. 14. 16:18
728x90

내가 처음에 짠 코드!!
왠지 스택으로 풀어야할것 같은 느낌이긴 했는데,,
편한 방법으로 풀어보려고 했던 나는 결국,,

def solution(ingredient):
    temp=[]
    answer =0
    for i in range(len(ingredient)):
            temp.append(ingredient[i])
            if temp[len(temp)-4:len(temp)+1] == [1,2,3,1]:
                answer+=1
                temp = temp[0:i-3]
    return answer

하지만 제출후 체점하기
눌렀더니
1,2 는 통과했지만
3번부터는 실패가 떴다..
왠지는 이유를아직 못찾음..
그래서 다시 스택으로 수정!

완성된 코드~~

def solution(ingredient):
    temp=[]
    answer =0
    for i in range(len(ingredient)):
            temp.append(ingredient[i])
            if temp[-4:] == [1,2,3,1]:
                answer+=1
                for k in range(4) :
                    temp.pop()
    return answer