(Python)백준 코딩테스트 연습 - 탑(2493)

1. 문제

https://www.acmicpc.net/problem/2493

2493번: 탑

2. 풀이

# 입력 코드
import sys
input = sys.stdin.readline

# n : 탑의 수
n = int(input())
# nList : 탑 크기 배열
nList = list(map(int, input().split()))

# 이전 탑의 크기를 비교하면서 자신의 크기보다 작으면 pop 크면 result에 앞 타워에 순서를 추가하고,
# 큰 타워가 없다면 0을 추가
result = ["0"]

tower = [[1, nList[0]]]

for k, v in enumerate(nList[1:]):
    while True:
        idx, length = map(int, tower.pop())
        if length >= v:
            result.append(str(idx))
            tower.append([idx, length])
            tower.append([k+2, v])
            break
    
        if len(tower) == 0:
            result.append("0")
            tower.append([k+2, v])
            break
            
print(" ".join(result))