[백준 알고리즘] 10989 수 정렬하기3 python

문제

N개의 수가 주어졌을 때, 이를 오름차순으로 정렬하는 프로그램을 작성하시오.

입력

첫째 줄에 수의 개수 N(1 ≤ N ≤ 10,000,000)이 주어진다. 둘째 줄부터 N개의 줄에는 수가 주어진다. 이 수는 10,000보다 작거나 같은 자연수이다.

출력

첫째 줄부터 N개의 줄에 오름차순으로 정렬한 결과를 한 줄에 하나씩 출력한다.

코드

import sys
n = int(input())
cnt = [0]*(10000+1)

for i in range(n):
    cnt[int(sys.stdin.readline())] +=1
for i in range(len(cnt)):
    for j in range(cnt[i]):
        print(i)

풀이

정렬 라이브러리를 사용하면 메모리 초과가 된다.

입력 데이터의 크기가 한정되어 있고, 데이터가 많이 중복되는 걸 봐서 계수정렬을 사용해 구현했다.

 

처음에, arr 배열에 append로 입력을 받고 cnt에 저장하는 방식으로 구현했는데 또 메모리 초과가 났다.

그래서

1) arr배열을 입력을 받을 때 바로 cnt의 index에 +1을 해주는 방식으로 구현했다.

2) cnt의 크기를 10000에서 +1 로 정해줬다.