글 읽기 - 파이썬 시간 초과 질문

dansan0303 2년 전 0

시간초과가 뜨는 이유를 모르겠습니다 ㅠㅠ

import sys
input = sys.stdin.readline
n = int(input())
lst = [int(input()) for _ in range(n)]
lst.sort()

#산술평균
print(round(sum(lst)/n))

#중앙값
print(lst[len(lst)//2])

#최빈값
setList = list(set(lst))
counts = [lst.count(i) for i in setList]
maximum = max(counts)
ans = []
for index,value in enumerate(counts):
    if value == maximum:
       ans.append(setList[index])
if len(ans) == 1:
    print(ans[0])
else:
    print(ans[1])
#범위
print(max(lst)-min(lst))

kong_22 2년 전 1

최빈값을 구할 때 counts = [lst.count(i) for i in setList]로 하셨는데요,

리스트의 count()메서드를 사용하면 lst 리스트를 모두 돌면서 원소의 갯수를 세줍니다.

이렇게하면 모든 원소 하나하나에 대해서 리스트를 한번씩 돌아야하므로 시간이 오래 걸립니다.

가령 [1,2,3,4] 리스트의 각 원소 개수를 센다면 1의 개수를 세는데 4번, 2의 개수를 세는데 4번,.. 총 16번의 원소 탐색이 필요합니다.

해결책으로는, 리스트를 한번만 돌면서 세주면 됩니다.

가령 첫번째 원소를 보고 1이니까 counts[1] += 1, 두번째 원소를 보고 2이니까 counts[2] += 1, ... 이런 식으로 셀 수 있다면 4번에 가능하다는 의미입니다.

dansan0303 2년 전 0

너무 감사합니다!!

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

파이썬 시간 초과 질문

dansan0303 2년 전 0

kong_22 2년 전 1

dansan0303 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말