[프로그래머스] 탑


스택/큐 - 탑

  • 문제 링크 :

사용 언어 : Python3

def solution(heights):
    tmp = []
    answer = []
    while(heights) :
        flag = 0
        right = heights.pop()
        for i in range(len(heights)-1, -1, -1) :
            if heights[i] > right :
                tmp.append(i+1)
                flag = 1
                break
        if flag == 0:
            tmp.append(0)
    answer = tmp[::-1]
    
    return answer


© 2020. by bs-derek