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

문제

N명의 사람들은 매일 아침 한 줄로 선다. 이 사람들은 자리를 마음대로 서지 못하고 오민식의 지시대로 선다.

어느 날 사람들은 오민식이 사람들이 줄 서는 위치를 기록해 놓는다는 것을 알았다. 그리고 아침에 자기가 기록해 놓은 것과 사람들이 줄을 선 위치가 맞는지 확인한다.

사람들은 자기보다 큰 사람이 왼쪽에 몇 명 있었는지만을 기억한다. N명의 사람이 있고, 사람들의 키는 1부터 N까지 모두 다르다.

각 사람들이 기억하는 정보가 주어질 때, 줄을 어떻게 서야 하는지 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 사람의 수 N이 주어진다. N은 10보다 작거나 같은 자연수이다. 둘째 줄에는 키가 1인 사람부터 차례대로 자기보다 키가 큰 사람이 왼쪽에 몇 명이 있었는지 주어진다. i번째 수는 0보다 크거나 같고, N-i보다 작거나 같다. i는 0부터 시작한다.

출력

첫째 줄에 줄을 선 순서대로 키를 출력한다.

===

풀이

import sys
input=sys.stdin.readline

n=int(input())
answer = [0 for _ in range(n)]
arr=list(map(int,input().split()))  #둘쨋줄에 입력받음

'''
arr[i]의 값만큼 왼쪽에 0이 있는 인덱스에 i+1을 넣자.
만약에 이미 자리가 있다면, 오른쪽으로 이동하자
'''

for i in range(n):

    findZero = 0 #왼쪽으로부터의 0의 개수
    idx = arr[i]
    wantIdx=0 #answer에 실제값을 넣을인덱스
    for j in range(n):
        if(findZero==idx):
            wantIdx=j# answer에 실제 값을 넣고자 하는 인덱스로 변경
            break
        if(answer[j]==0):
            findZero+=1 
    while(answer[wantIdx]!=0):
        wantIdx+=1
    answer[wantIdx] = i+1
print(*answer)

아이디어가 있으면 구현하는 데에 큰 어려움은 없는 문제이다.
예를들어 n이 5이고
arr[0]가 2라면,(키가 1인사람)
0 0 1 0 0
arr[1]가 3이라면 (키가 2인사람)
0 0 1 0 2
arr[2]가 2라면 (키가 3인사람)
0 0 1 3 2
.... (왼쪽에 비워둔 자리는 키가 더 큰 사람이 설 자리)

이런 식으로 정답 배열을 만들면 된다.