백준 1620번 나는야 포켓몬 마스터 이다솜 파이썬

 

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

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

 

내 답안

import sys

input = sys.stdin.readline
N, M = map(int, input().rstrip().split())
x = [input().rstrip() for i in range(N)]
y = dict()

for i in range(N):
    y[x[i]] = str(i + 1)

z = dict(map(reversed, y.items()))

for i in range(M):
    a = input().rstrip()
    if a.isdigit():
        print(z[a])
    else:
        print(y[a])

---

 

y 는 {'포켓몬 이름' : '번호'} 로 되어 있는 딕셔너리이고

z 는 {'번호' : '포켓몬 이름'} 로 되어있는 딕셔너리이다.

 

z = dict(map(reversed, y.items()))

11번째 줄 코드가 key 와 value를 바꿔주는 코드다.

알파벳으로 입력받으면 y에서 값을 찾아주고

알파벳이 아닌 문자열로 입력받았다면 z에서 값을 찾아주는 코드를 짰다.