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

메모리 초과로 고생을 좀 했다...

문제

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

풀이

• 메모리 제한이 8MB로 리스트와 힙정렬이나 퀵정렬과 같은 일반적인 정렬 알고리즘을 사용했을 때 메모리 초과가 떴다.
• Counting sort를 사용해서 입력 받은 nums 이외에 counting에 필요한 메모리만 사용하도록 하여 메모리 사용량을 줄여봤다.
  • 하지만 계속해서 메모리 초과 오류가 떠서 메모리 사용을 줄일 수 있는 곳을 찾아보니 nums를 초기화할 때 list comprehension을 사용하지 않고 generator 표현을 사용하면 메모리 사용을 줄일 수 있다는 것을 찾게 되었다.

    import sys

    input = sys.stdin.readline
    N = int(input())
    # list comprehension 사용으로 리스트를 메모리에 저장하게 됨
    nums = [int(input()) for _ in range(N)]

    # 문제 조건을 참고하여 counts 리스트 선언
    counts = [0] * 10001
    for n in nums:
        counts[n] += 1

    # counting sort
    for i, count in enumerate(counts):
        while count:
            print(i)
            count -= 1

• 아래와 같이 6번째 줄을 generator 표현으로 바꿈으로써 입력 받은 숫자들을 모두 메모리에 올리지 않고 iteration을 돌 수 있도록 하여 문제를 해결하였다.

    ...
    # generator를 통해서 리스트를 메모리에 두지 않고 nums 입력
    nums = (int(input()) for _ in range(N))
    ...
  

• 앞으로는 파이썬을 사용할 때 실행속도도 항상 염두에 두고 사용해야겠고 느꼈다.

참고자료

• 파이썬 실행속도 개선 방법