한 줄로 서기

박고은·2023년 7월 4일
0

코딩테스트 연습

목록 보기
24/34

문제 ✨

1138번: 한 줄로 서기

N명의 사람들은 매일 아침 한 줄로 선다. 이 사람들은 자리를 마음대로 서지 못하고 오민식의 지시대로 선다.
어느 날 사람들은 오민식이 사람들이 줄 서는 위치를 기록해 놓는다는 것을 알았다. 그리고 아침에 자기가 기록해 놓은 것과 사람들이 줄을 선 위치가 맞는지 확인한다.
사람들은 자기보다 큰 사람이 왼쪽에 몇 명 있었는지만을 기억한다. N명의 사람이 있고, 사람들의 키는 1부터 N까지 모두 다르다.
각 사람들이 기억하는 정보가 주어질 때, 줄을 어떻게 서야 하는지 출력하는 프로그램을 작성하시오.

N = int(input())
row = list(map(int, input().split()))
answer = [0]*N

for i in range(N):
    zero = len([n for n in answer[:row[i]] if n==0])
    for j in range(row[i], N):
        if answer[j]==0:
            if zero==row[i]:
                answer[j] = i+1
                break
            else:
                zero += 1
                continue

print(*answer)

0으로 된 배열을 생성하고 키가 작은 순서대로 배치하면 매 순서마다 0의 개수는 배치하는 사람의 키보다 큰 사람의 수와 일치

0개의 댓글