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

문제 링크

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

10989번: 수 정렬하기 3

 

문제 풀이

이 문제는 메모리 할당에 주의해서 풀어야 했다. 숫자를 입력 받아서 정렬 함수를 사용하는 것으로는 메모리 초과가 되기 때문이다. 문제를 보면, 10,000보다 작거나 같은 자연수라고 되어있다. 이에 따라 처음부터 저 만큼의 메모리를 할당해두고, 입력되는 주어지는 수들의 개수를 카운트하여 해결하였다.

 

전체 코드

from sys import stdin
input = stdin.readline

nums = [0 for _ in range(10001)]
n = int(input())
for _ in range(n) :
    x = int(input())
    nums[x] += 1

for i in range(len(nums)) :
    if nums[i] != 0 : 
        for _ in range(nums[i]) :
            print(i)
    else : continue

 

메모리 초과된 코드..

 

from sys import stdin
input = stdin.readline

n = int(input())
l = [input().strip() for _ in range(n)]
l.sort()

for i in l :
    print(i)