알고리즘/기타

[백준 10989] 수 정렬하기 3 (파이썬 풀이)

파프리카. 2023. 2. 13. 19:05

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

 

10989번: 수 정렬하기 3

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

www.acmicpc.net

문제

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

제출 코드 1 (메모리 초과)

n = int(input())
l = []
for _ in range(n):
    l.append(int(input()))
l.sort()
for i in l:
    print(i)

for문 안에서 append를 사용하면 매 스텝마다 메모리 재할당이 일어난다. 비효율적인 코드다.

제출 코드 2

import sys
input = sys.stdin.readline
n = int(input())
l = [0] * 10001
for _ in range(n):
    l[int(input())] += 1
for i in range(10001):
    if l[i] != 0:
        for _ in range(l[i]):
            print(i)

1. input 대신 readline을 쓰면 시간 단축된다.

2. 먼저 0으로 채워진 리스트를 만들고 입력값을 인덱스로 받아 메모리 할당을 줄였다.