문제
N개의 수가 주어졌을 때, 이를 오름차순으로 정렬하는 프로그램을 작성하시오.
입력
첫째 줄에 수의 개수 N(1 ≤ N ≤ 1,000,000)이 주어진다. 둘째 줄부터 N개의 줄에는 수가 주어진다. 이 수는 절댓값이 1,000,000보다 작거나 같은 정수이다. 수는 중복되지 않는다.
출력
첫째 줄부터 N개의 줄에 오름차순으로 정렬한 결과를 한 줄에 하나씩 출력한다.
내 제출
해설
처음으로 작성한 코드이다.
n = int(input())
num = []
for i in range(n):
x = int(input())
num.append(x)
for i in sorted(num):
print(i)이 코드를 이용하면 Python3에서는 시간 초과가 발생하지만 PyPy3에서는 성공한다.
PyPy3는 JIT 컴파일을 도입해 반복문 등 복잡한 코드를 실행할 때 캐싱을 통해 시간을 단축시킨다.
성공했지만 Python3로도 성공할 수 있도록 sys를 import 시켰다.
참고내용 : https://ralp0217.tistory.com/entry/Python3-%EC%99%80-PyPy3-%EC%B0%A8%EC%9D%B4
import sys
n = int(sys.stdin.readline())
num = []
for i in range(n):
num.append(int(sys.stdin.readline()))
for i in sorted(num):
sys.stdout.write(str(i)+'\n')이렇게 하면 성공이 뜬다.
여기서 궁금해진 게 input과 print에서 sys를 사용해서 통과했는데, 그러면 input() 함수와 sys.stdin의 시간 차이는 얼마나 날까?
해서 input 함수만 sys를 이용하고 print는 그대로 적용한 코드를 채점해보았다.
약 100ms 차이가 난다.
문제를 풀 때는 매우 귀중한 시간이니까 시간 초과가 발생한다면 sys를 import 했는지 살펴보면 좋겠다.