[알고리즘 풀이] 프로그래머스 : 탑, Level2(스택/큐)

프로그래머스(https://programmers.co.kr/)

프로그래밍 강의 | 프로그래머스

프로그래머스 코딩테스트 중 Level 2 탑 문제 푼 내용입니다. 

Python 3 으로 풀어봤습니다.

 

python 리스트의 기본 method인 pop, insert를 사용해서

리스트를 스택(stack)으로 사용해 문제를 풀었습니다.

 

def solution(heights):
    
    s_result = [] # 마지막 출력용
    
    while len(heights) > 0 :
        h_temp = heights.pop()
        # print(heights)
        s1 = [i for i in heights]
        for i in range(len(s1)) :
            if s1.pop() > h_temp :
                s_result.insert(0,len(s1)+1)  
                break
            elif len(s1) == 0 :
                s_result.insert(0,0)
    s_result.insert(0,0)
    return s_result