글 읽기 - 파이썬으로 시간초과

oyhskw 4년 전 0

우선 저는 입력으로 숫자와 문자열이 들어오는 경우를 나눠서 생각해봤습니다.

처음에 포켓몬 이름을 순서대로 입력받을 때, 리스트에 차곡차곡 포켓몬 이름을 넣고, 동시에 포켓몬 이름을 키로 갖고 값으로 인덱스를 갖는 딕셔너리를

이용했습니다.

1) 숫자 인 경우

리스트로 저장된 포켓몬 이름을 숫자 인덱스를 통해 구했습니다.

2) 문자열인 경우

딕셔너리의 키 값을 통해 인덱스를 구했습니다.

요렇게 했는데 시간초과가 뜨네요; 근데 이러한 접근 방식이 너무 비효율적인가요??

# 백준 - 자료구조 - 나는야 프켓몬   마스터 이다솜
def isNumber(s):
    try:
        int(s)
        return True
    except ValueError:
        return False

poket_mon_num, need_num = map(int, input().split())
poket_mon_list = []
poket_mon_dic = {}

for i in range(0, poket_mon_num):
    poket_mon_name = input()
    poket_mon_list.append(poket_mon_name)
    poket_mon_dic[poket_mon_name] = i+1

for i in range(0, need_num):
    ask = input()

    if isNumber(ask):
        ask = int(ask)
        answer = poket_mon_list[ask - 1]
    else:
        answer = poket_mon_dic[ask]

    print(answer)

wjsqjawns 4년 전 0

입력을 받는 게 느려서 그렇네요.

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

readline()을 사용하시면 정답 처리가 됩니다.

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

문제

문제

출처

ICPC

파이썬으로 시간초과

oyhskw 4년 전 0

wjsqjawns 4년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말