[BaekJoon] 1620 나는야 포켓몬 마스터 이다솜

▶ 문제 :

 

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

 

▶ 코드 :

 

n, m = map(int, input().split())

numToName = {}
nameToNum = {}
question = []

nums = ['0','1','2','3','4','5','6','7','8','9']

for i in range(1,n+1):
    data = input()
    numToName[i] = data
    nameToNum[data.lower()] = i

for i in range(m):
    question.append(input().lower())

for q in question:
    if q[0] in nums:
        print(numToName[int(q)])
    else:
        print(nameToNum[q.lower()])

 

▶ 문제 풀이 : 

 

1. 초기 입력을 받아 각각 키, 값을 이름 : 번호, 번호 : 이름 을 가지는 Dictionary 2개를 생성하였다. 

2. 이후 각각의 question이 숫자가 포함된 경우와 아닌 경우를 판단해 Dictionary 참조를 해주었다.

 

=> Dictionary는 해싱을 사용해 참조 속도가 빠르기에 이용하였다.