[백준] 18110번: solved.ac 파이썬

https://www.acmicpc.net/problem/18110

18110번: solved.ac

왜 시간초과지?

 

간단한 문제라고 생각해서 바로 코드 짜서 돌렸더니 

시 간 초 과

 

오잉? 시간복잡도도 괜찮은데 왜지?

 

이유는 input() 때문이었다.

입력으로 주어지는 숫자 개수가 10의5승.........짱많음

 

이런 경우엔 input()말고 sys.stdin.readline()을 써야 빠른 입력을 할 수 있다.

 

풀이

import math, sys
n = int(sys.stdin.readline())

if(n == 0):
  print(0)
else:
  arr = [0] * n
  check = math.floor(n*0.15+0.5)
  sum = 0

  for i in range(n):
    arr[i] = int(sys.stdin.readline())

  arr.sort()

  for i in range(check, n - check):
    sum += arr[i]

  average = math.floor(sum / (n - 2*check) + 0.5)
  print(average)