스택활용법 : 파이썬

u·2022년 4월 30일
0

Algorithm

목록 보기
17/21

스택을 특정한 조건에 따라 활용해야 할 때 아래와 같이 쓸 수 있다.


def solution(list, k):
    stack = []
    
    for item in list:
        while stack!=[] and stack[-1]<n and k>0:
            stack.pop()
            k -= 1
        stack.append(item) 
   
   return stack

스택을 이용할 때 위 코드처럼 while문을 통해 특정 조건에서 pop해가면서 stack을 완성하면된다

0개의 댓글