[Python] 백준 10989번 : 수 정렬하기 3

문제

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

입력

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

출력

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

예제 입력 1

10
5
2
3
1
4
2
3
5
1
7

예제 출력 1

1
1
2
2
3
3
4
5
5
7

 

소스코드

import sys

n = int(sys.stdin.readline())
li = [0] * 10001

for i in range(n):
    num = int(sys.stdin.readline())
    li[num] += 1

for i in range(1, len(li)):
    if li[i] != 0:
        for j in range(li[i]):
            print(i)

 

1. 처음에는 input()을 이용하여 값을 받아왔는데, input()이 메모리를 많이 사용한다는 것을 알게 되어 input()을 전부 

import sys

sys.stdin.readline()

으로 바꿔주었다.

 

2. 입력 방식을 바꿨음에도 메모리 초과가 나서 입력값을 저장하는 방식을 바꿔보았다. 처음에는 입력값을 append를 이용해 리스트에 하나씩 넣어주었는데, 미리 0으로 초기화된 10001 크기의 배열을 만들고, 입력값을 인덱스로 하는 배열값을 1 증가시키는 것으로 바꿔주었다.

 

위 두가지 방식을 이용해서 메모리 초과를 해결할 수 있었던 문제이다.