글 읽기 - 시간초과가 어디서 뜨는지 모르겠어요

kkumiya 1년 전 0

입력받은 m을 given_list에 넣고

elem이 elem.upper()이랑 같으면 숫자로 판단해서 int(elem)으로 숫자로 변환하고 출력하고

elem이 elem.upper()이랑 다르면 문자열로 판단해서 index사용해서 출력하는 방법인데

pypy로하면 바로 맞았습니다가뜨는데 python으로하면 시간초과가뜨네요

어디서 시간초과가 뜨는건가요?

#백준 1620번-나는야 포켓몬 마스터 이다솜

import sys

n, m = map(int, sys.stdin.readline().split())
arr = []
for _ in range(n):
    poketmon = sys.stdin.readline().strip()
    arr.append(poketmon)

given_list = []
for _ in range(m):
    x = sys.stdin.readline().strip()
    given_list.append(x)

for elem in given_list:
    if elem == elem.upper():    #숫자
        elem = int(elem)
        print(arr[elem-1])
    else:                       #문자열
        print(arr.index(elem)+1)

dong5995 1년 전 0

index로 찾는 게 아마 느릴 거에요.

딕셔너리 쓰면 더 빨라요

kkumiya 1년 전 0

감사합니당

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

문제

문제

출처

ICPC

시간초과가 어디서 뜨는지 모르겠어요

kkumiya 1년 전 0

dong5995 1년 전 0

kkumiya 1년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말