글 읽기 - 파이썬 시간초과 질문드립니다 ㅠㅠ

salvasoo235 2년 전 0

최빈값에서 계속 시간초과가 뜨는 것 같은데요..ㅜ

포문 하나 썼으니 복잡도는 O(n) 아닌가요?? 8000+n1+n2 이런식으로 될 것 같은데요 시도 횟수가..

고수님들 고견을 구합니다..

import math
import sys

def find_mod(n): #최빈값 구하기

    count=0
    num=list()
    a=[0]*8001
    for i in range(len(n)):
        a[4000+n[i]]+=1 #0~8000을 각 자리에 저장, 해당되는 값의 자리수에 count


    b=a.index(max(a)) #주소 저장
    c=max(a) #max값 저장
    a[b]=0 #max값 날리고

    if c!=max(a): #max 날렸을 때 이전과 값이 다르다면
        return b-4000 #저장해놓은 index값에 -4000을 빼서(최빈값이 단 1개였으므로) 그 값을 반환

    else:
        return a.index(max(a))-4000 #두번째로 작은, 즉 앞에서 삭제한 후의 index 값을 반환

T=int(input())
T_list=list()
for i in range(T):
    T_list.append(int(input()))

T_list_sort=T_list
T_list_sort.sort() #정렬

print(round(sum(T_list)/len(T_list)))
print(T_list_sort[math.trunc(len(T_list)/2)])
print(find_mod(T_list)) #최빈값 구하기
print(T_list_sort[len(T_list_sort)-1]-T_list_sort[0])

ujoon 2년 전 1

input()문 보다 sys.stdin.readline()으로 읽어드리는 것이 더 빠릅니다. 아마도 코드가 잘못되지 않았다면 해결 될 수도 있지 않을까 합니다.

아니라면 pypy3으로도 제출해봐도 좋을 듯 합니다.

salvasoo235 2년 전 1

맞네요 선생님... readline 썼더니 바로 통과되었습니다

이게 차이가 큰가보네요 항상 그냥 input()으로 썼는데..

감사합니다

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

문제

문제

출처

ICPC

파이썬 시간초과 질문드립니다 ㅠㅠ

salvasoo235 2년 전 0

ujoon 2년 전 1

salvasoo235 2년 전 1

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말