[백준-1138] 한 줄로 서기 / Python

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

1138번: 한 줄로 서기

import sys

input = sys.stdin.readline

n = int(input())
line = list(map(int, input().split()))

answer = [0] * n

for i in range(1, n + 1):
    temp = line[i - 1]
    cnt = 0
    for j in range(n):
        if temp == cnt and answer[j] == 0:
            answer[j] = i
            break
        elif answer[j] == 0:
            cnt += 1

print(*answer)

 

해설

answer에 값이 들어갈 인덱스 앞 부분의 0들은 모두 그 값보다 큰 값이 들어가게 된다.

 

cnt는 answer를 앞에서부터 탐색했을 때 0의 개수이다.

line[i-1]이 cnt와 같을 때 바로 다음 answer 인덱스에 i를 넣어준다. 이미 answer의 해당 인덱스에 값이 들어간 경우라면 계속 리스트를 돌아 0인 부분에 i를 넣어준다.