[알고리즘/백준] 1138번 : 한 줄로 서기(python)

키가 1인 사람부터 주어지기 때문에 반복문을 사용하여 돌려준다. 1의 위치를 찾았으면 순서대로 2 3 4 ... 찾으면 된다.

N = int(input())
a = list(map(int, input().split()))
b = [0 for _ in range(N)]
for i in range(N):
    t = a[i]
    cnt = 0
    for j in range(N):
        if cnt == t and b[j] == 0:
            b[j] = i + 1
            break
        elif b[j] == 0:
            cnt += 1
print(*b)

좋은 웹페이지 즐겨찾기