[Programmers] 스택/큐 : 탑 (Python)

1일 1문제 13일차!

오늘의 문제는 프로그래머스 스택/큐에있는 탑 문제입니다.

 

코딩테스트 연습 - 탑 | 프로그래머스

def solution(heights):
    answer = []
    
    for i in range(len(heights)):
        if i == 0:
            answer.append(0)
        else:
            for j in range(i, 0, -1):
                check_flag = False
                if heights[i] < heights[j]:
                    answer.append(j+1)
                    check_flag = True
                    break
            if check_flag == False:
                if heights[0] > heights[i]:
                    answer.append(1)
                else:
                    answer.append(0) 
                    
    return answer