아래 백준 로고를 클릭하면 해당 문제로 이동합니다 😀
풀이과정
일단 pocketmon 딕셔너리를 만들고 고유번호에 동물을 매칭했다.
그리고 M개를 입력받을 때, isalpha()를 통해 알파벳인지 숫자인지 구분해줬다.
근데 여기서 ... 알파벳이면 딕셔너리에서 key를 바로 빼오면 되는데 숫자인 경우는 숫자가 value니까 어떻게 찾아내지 ... ? 를 생각했는데 ...
그냥 처음에 딕셔너리에 넣을때 key:value 쌍을 value:key 쌍으로 저장한 새로운 딕셔너리를 만들어버렸다 ^^...
그래서 알파벳이면 pocketmon에서, 숫자라면 pocketmon_num에서 value값을 가져오게 했다.
## 코드 --- ```python import sys
N, M = map(int, input().split())
pocketmon = {}
pocketmon_num = {}
for i in range(N):
name = sys.stdin.readline().rstrip()
pocketmon[name] = i+1
pocketmon_num[i+1] = name
for i in range(M):
order = sys.stdin.readline().rstrip()
if order in pocketmon.keys():
print(pocketmon[order])
else:
print(pocketmon_num[int(order)])
<br>
<br>
## 결과
---
