[백준 10989번] 수 정렬하기3 파이썬 python

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

10989번: 수 정렬하기 3

 

문제

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

시간	제한메모리
5 초	8 MB

입력

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

출력

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

---

해결법

일단 시간이랑 메모리 제한이 엄청 심해서 초기화를 할 때 수를 맞춰해야했다.

처음 count 초기화 때 N+1개를 했는데 계속 메모리 초과가 났다.

문제를 잘 보면 수는 10000보다 작거나 같은 수가 입력되므로 index를 수로 하고 값을 count로 하는 count 배열을 10001개로 초기화 한다.

다음과 같이 하니 메모리,시간 안에 결과가 나왔다.

import sys; input=sys.stdin.readline

N = int(input())

count =[0]*(10001) #N+1 대신 10001
for _ in range(N):
    a = int(input())
    count[a]+=1

for n in range(1,10001):
    if count[n]>0:
        for _ in range(count[n]):
            print(n)

 

배운점

앞으로 시간초과를 막기 위해 초기화시 최대 개수를 고려해야겠다.

또 정렬할때 배열에 입력해서 순서대로 뽑으면 sort보다 적은 시간이 걸림을 확인한다.