[baekjoon] 백준 1138번(파이썬): 한 줄로 서기

문제

1138번: 한 줄로 서기

알고리즘

- 반복문을 통해 각 자리에 들어갈 사람을 확인한다.

- 반복문을 통해 각 사람의 왼쪽에 키 큰 사람의 수를 확인한다.

- 자신의 왼쪽에 키 큰 사람의 수가 카운트한 수와 맞고 그 자리에 아무도 없다면 그 자리는 그 사람의 자리다.

- 자신의 왼쪽에 키 큰사람의 수가 카운트한 수가 아니고 자리에 아무도 없다면 자신의 왼쪽에 키 큰 사람의 수를 카운트한다.

- 자신의 왼쪽에 키 큰 사람의 수가 카운트한 수가 아니고 자리에 있는 경우는 자신보다 작은 사람이므로 카운트하지 않는다.

코드

import sys

n = int(sys.stdin.readline())
m = list(map(int, sys.stdin.readline().split()))
temp = [0] * n

# 반복문을 통해 각 자리에 들어갈 사람을 확인
for i in range(n):
    cnt = 0 # 자신의 왼쪽에 키 큰 사람의 수

    # 반복문을 통해 모든 사람을 확인
    for j in range(n):
        # 자신의 왼쪽에 키 큰 사람의 수가 맞고 그 자리에 아무도 없다면
        if cnt == m[i] and temp[j] == 0:
            temp[j] = i + 1
            break

        # 자리에 아무도 없다면 자신의 왼쪽에 키 큰 사람의 수를 카운트
        elif temp[j] == 0:
            cnt += 1

print(*temp)

github

GitHub - junjange/CodingTest: 내가 푼 코딩 테스트 문제와 해결법