알고리즘/기타
[백준 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으로 채워진 리스트를 만들고 입력값을 인덱스로 받아 메모리 할당을 줄였다.