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

✔ 문제

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

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

---

✔ 풀이

import sys
input = sys.stdin.readline

n, m = map(int, input().split())
poket = {}

for i in range(1, n+1):
    x = input().rstrip()
    poket[i] = x
    poket[x] = i

for j in range(m):
    x = input().rstrip()

    if x.isdigit() == True: # 입력 값이 숫자인지
        print(poket[int(x)])
    else:
        print(poket[x])

---

✔ 설명

key, value 동시에 값을 저장할 수 있는 dictionary를 사용하면 쉽게 풀 수 있다.

값을 입력받아서 key:value를 번호:이름, 이름:번호로 저장한 후 해당 입력 값으로 원하는 값을 뽑아 출력한다.

---

✔ 후기

input().rstrip()을 안했더니 오류가 떠서 rstrip()을 추가해줬다.

입력 값이 숫자인지 문자인지 판별하는 if문에서 잠깐 멈칫했다. 입력 값 길이로 판별해야하나 했지만 예제서부터 포켓몬 수가 10마리 넘어가기 때문에 isdigit() 함수를 겨우 기억해내서 사용했다.