[백준/Python]1620_나는야 포켓몬 마스터 이다솜

문제

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

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

풀이

특정값 탐색에 유리한 해시 테이블 자료구조를 사용합니다.

해시 테이블에서 특정값 탐색의 시간 복잡도는 O(1)입니다.

파이썬에선 딕셔너리 타입으로 이를 구현합니다.

isalpha()는 해당 문자열이 한글이나 알파벳으로만 이루어져있는지 확인하는 함수입니다.

isdecimal()는 해당 문자열이 숫자로만 이루어져있는지 확인하는 함수입니다.

코드

import sys

n, m = map(int, sys.stdin.readline().rstrip().split())
dictionary1 = dict()
dictionary2 = dict()
question = []
for i in range(1, n + 1):
    name = sys.stdin.readline().rstrip()
    dictionary1[i] = name
    dictionary2[name] = i
for _ in range(m):
    question.append(sys.stdin.readline().rstrip())
for q in question:
    if q.isalpha():
        print(dictionary2[q])
        continue
    if q.isdecimal():
        print(dictionary1[int(q)])