1138

날짜 : 2023. 08. 07

사용 언어 : python

 

문제

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

 

 

코드

import sys

n = int(sys.stdin.readline())
a = list(map(int, sys.stdin.readline().split()))

res = [0] * n
res[a[0]] = 1

for i in range(2, n+1):
    cnt = 0  # 자기보다 큰 사람의 수
    idx = 0  # 서있는 위치
    while True:
        if res[idx] == 0:
            if a[i-1] == cnt:
                res[idx] = i
                break
            else:
                idx += 1
                cnt += 1
        else:
            idx += 1
    # print(*res)

print(*res)

 

 

풀이

- 키가 1인 경우에는 무조건 본인보다 큰 사람이 없기 때문에 디폴트로 값을 지정한다.

- 그 다음 키가 2인 사람부터는 본인보다 큰 사람의 수와 인덱스를 함께 비교하면서 while문을 통해 올려나간다

- res에서 0이 아닌 경우는 이미 설정이 되어있기 때문에 pass

- 이후 리스트 a에서의 값들은 해당 키 기준 왼쪽에 더 장신인 사람들의 수이기 때문에 idx, cnt를 올려나간다.

- 마지막으로 결과값이 비어있고, 리스트 a의 값과 cnt값(비교했을 때 더 큰 사람들 수)이 동일한 경우 이를 설정한다.

 

 

알게된 점

- idx 하나로 구현하려 했지만 계속 인덱스 에러가 발생했고, 결과값은 비어있지만 큰 수를 가산하지 않은 케이스를 간과해서 계속 에러가 발생한 것 같다.

- idx, cnt 두개의 변수를 사용하니까 문제는 해결됐다.

 

 

참고 사이트

-