📍 백준 #10989

❓ 오류 원인 찾기

n = int(input())

array = []
for i in range(n):
  array.append(int(input()))

array.sort()

for data in array:
  print(data)

처음 위와 같은 코드를 제출했는데 메모리 초과 오류가 발생했다.
오류 발생 이유를 찾아본 결과 이전에 풀었던 문제와는 달리 메모리 제한이 8MB로 되어있었다.
array에 데이터를 하나씩 추가하고 sort()를 하는 과정에서 메모리 초과가 발생한 것이다.

❗ 계수 정렬 사용

오류 해결을 위해 계수 정렬 방식을 사용했다.

n = int(input())

count = [0] * 10001

for i in range(n):
  num = int(input())
  count[num] += 1

for i in range(len(count)):
  for j in range(count[i]):
    print(i)

count라는 리스트를 입력 데이터의 가장 큰 수보다 하나 크게 (인덱스 0은 사용하지 않기 위해) 만든다.
데이터를 하나씩 입력받으면서 count 리스트의 데이터와 같은 인덱스의 수를 하나씩 증가시킨다.
반복문을 통해 count 리스트의 각각의 원소 값 만큼 내부 반복문을 돌려 인덱스를 출력한다.

❓ 이 코드는 시간 초과 오류가 발생했다.

🌟 시간 초과 오류 해결 -> sys.stdin.readline()

import sys

n = int(input())

count = [0] * 10001

for i in range(n):
  num = int(sys.stdin.readline())
  count[num] += 1

for i in range(len(count)):
  for j in range(count[i]):
    print(i)

input()을 이용해 입력을 받으면서 시간 초과가 발생했기 때문에 sys 라이브러리의 sys.stdin.readline()함수를 이용해 시간을 단축했다.

📢 sys.stdin.readline()

import sys

input_data = sys.stdin.readline().rstrip()

print(input_data)

sys : 파이썬 인터프리터가 제공하는 변수와 함수를 직접 제어할 수 있도록 해주는 라이브러리
sys.stdin.readline() : 한 줄 단위로 입력 되며 개행문자가 같이 입력된다. 공백 문자를 제거하려면 rstrip()함수를 사용한다.