[프로그래머스] 배열 만들기 4 - 파이썬

Programmers

배열 만들기 4

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

프로그래머스

문제

코드

<풀이>

def solution(arr):
    stk = []
    i = 0
    while i < len(arr):
        if len(stk) == 0: 
            stk.append(arr[i])
            i += 1
        elif len(stk) > 0 and stk[-1] < arr[i]:
            stk.append(arr[i])
            i += 1
        elif len(stk) > 0 and stk[-1] >= arr[i]:
            stk.pop()
    return stk

 

<다른 풀이>

def solution(arr):
    stk = []
    for i in range(len(arr)):
        while stk and stk[-1] >= arr[i]:
            stk.pop()
        stk.append(arr[i])
    return stk

Reference

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