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

 

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

dic_n = {}
dic_s = {}
for i in range(n):
    v = input().strip()
    dic_n[str(i+1)] = v
    dic_s[v] = i+1

for i in range(m):
    q = input().strip()
    if q.isdigit():
        print(dic_n[q])
    if q.isalpha():
        print(dic_s[q])

 

딕셔너리 2개를 사용해야 한다

{숫자:이름} : 숫자로 문자 출력

{이름:숫자} : 문자로 숫자 출력

input().strip()을 사용한 건 \n 마지막 공백 문자를 제거해 주기 위해서

isdigit()로 숫자인지 판별

isalpha()로 영문이나 한글인지 확인