본문 바로가기
알고리즘/BOJ

[BOJ / 백준] 1620번 나는야 포켓몬 마스터 이다솜 python / 파이썬 딕셔너리 /Key값으로 Value찾기

by seohmoon 2022. 4. 30.

출처 : https://www.acmicpc.net/problem/1620

 

1620번: 나는야 포켓몬 마스터 이다솜

첫째 줄에는 도감에 수록되어 있는 포켓몬의 개수 N이랑 내가 맞춰야 하는 문제의 개수 M이 주어져. N과 M은 1보다 크거나 같고, 100,000보다 작거나 같은 자연수인데, 자연수가 뭔지는 알지? 모르면

www.acmicpc.net


딕셔너리로 안 풀고 리스트로 풀었더니 시간 초과 나왔다.

딕셔너리 두개 만들어서 숫자일 때, 문자일 때 나눴다.

import sys
N, M = map(int, input().split())
pokemon = {}
pokemon2 = {}
for i in range(1, N+1):
    pokemonnumber = str(i)
    pokemonname = sys.stdin.readline().strip()
    pokemon[pokemonname] = pokemonnumber
    pokemon2[pokemonnumber] = pokemonname

for j in range(M):
    j = sys.stdin.readline().strip()
   
    if 49 <= ord(j[0]) <= 57: # 숫자일 때
        print(pokemon2[j])
    else:
        print(pokemon[j])

아스키코드 값으로 48부터 안한건 앞자리가 0이 올 수는 없어서 49부터 했다.

댓글