백준 2108번 통계학 파이썬

백준 2108번 통계학 파이썬 코드이다.

from sys import stdin

a = int(stdin.readline())
b = []
for _ in range(a):
    b.append(int(stdin.readline()))
print(round(sum(b) / a))
b = sorted(b)
print(b[a // 2])
x = {}
for i in b:
    if f"{i}" not in x:
        x[f"{i}"] = 0
    x[f"{i}"] += 1
m = max(x.values())
result = []
for k, v in x.items():
    if len(result) > 2:
        break
    if v == m:
        result.append(int(k))
if len(result) > 1:
    print(result[1])
else:
    print(result[0])
print(b[a - 1] - b[0])

입력을 input()으로 받으면 시간초과가 떠서 stdin.readline()으로 변경 후 통과할 수 있었다.

반복되는 입력을 많이 받을 문제일 경우, stdin.readline()으로 입력을 받으면 시간을 단축할 수 있다.