[백준] 2750-2751번 수 정렬하기(Python)/ sys.stdin.readline() 오류 해결

📌 2750번 - 수 정렬하기

2750번: 수 정렬하기

💻 소스 코드

N = int(input())

arr = []
for i in range(N):
    num = int(input())
    arr.append(num)

for i in range(N-1):
    min_idx = i
    for j in range(i+1, N):
        if arr[min_idx] > arr[j]:
            min_idx = j
    arr[i], arr[min_idx] = arr[min_idx], arr[i]

for i in arr:
    print(i)

💡 문제 해결

수를 입력받아 리스트에 넣은 후, 이중 for문을 돌며 두 요소씩 비교합니다. 그리고 더 작은 값이 나타날 때마다 현재 값보다 앞으로 순서를 옮겨줍니다.
이 문제는 입력받는 수의 개수 N의 범위가 1 ≤ N ≤ 1,000이기 때문에 별다른 문제 없이 해결됩니다. 하지만 N의 범위가 커지면 문제가 생깁니다. 예시로 _"2751번 수 정렬하기 2"_를 살펴보겠습니다.

📌 2751번 - 수 정렬하기 2

2751번: 수 정렬하기 2

💻 소스코드

# 입력 횟수가 많아지면 빠른 입출력 필요 --> sys.stdin.readline() 이용

import sys

N = int(sys.stdin.readline())
num_list = []

for i in range(N):
    num_list.append(int(sys.stdin.readline()))

num_list = sorted(num_list)
for i in num_list:
    print(i)

💡 문제 해결

문제 해결 방식은 _"2750번 - 수 정렬하기"_와 다르지 않습니다. 2750번 소스 코드를 그대로 써도 답은 잘만 나옵니다. 그러나 백준에 그대로 제출하면 '시간 초과'가 뜰 것입니다.
이 문제는 N의 범위가 1 ≤ N ≤ 1,000,000이기 때문에 입력 횟수를 최대까지 받을 경우 시간이 오래 걸립니다. 그래서 빠른 입출력이 필요한데, 이때 sys.stdin.readline()을 쓸 수 있습니다. 또한 이중 for문으로 정렬하는 대신, sorted() 함수를 써서 시간을 더 줄여줬습니다.

🚨 주의사항

jupyter에는 stdin이 제대로 구성되어 있지 않기 때문에
sys.stdin.readline()를 사용하면 제대로 입력받지 못합니다.
마찬가지로, jupyter notebook (.ipynb)파일에서도 제대로 작동하지 않습니다.

그대로 백준에 제출해도 성공은 가능합니다.

그러니 주피터나 주피터 노트북으로 작성하여 입력에 문제가 생긴 분들은
파이썬 (.py) 파일로 실행하면 문제가 해결될 겁니다.