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

1. 문제와 예제

(문제 링크: https://www.acmicpc.net/problem/1620)

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

문제 (문제 길이가 길어서 생략)

예제

---

2. 전체 코드

import sys
input = sys.stdin.readline

# 입력 받기
N, M = map(int, input().split()) # 도감에 있는 포켓몬 수, 맞춰야 하는 문제 수
dct = {} # dict

# {'포켓몬 번호': '포켓몬 이름', '포켓몬 이름': '포켓몬 번호'} 형식의 dict 생성
for i in range(N): # i: 포켓몬 번호
    name = input().rstrip() # name: 포켓몬 이름
    dct[i] = name
    dct[name] = i

for _ in range(M):
    que = input().rstrip() # 문제
    if que.isdigit(): # 문자가 들어왔으면
        print(dct[int(que)-1]) # 포켓몬 번호 출력
    else: # 숫자가 들어왔으면
        print(dct[que]+1) # 포켓몬 이름 출력

---

3. 코드 해설

문제로 포켓몬 이름(문자)가 들어오면 포켓몬 번호를 출력해야 하고, 반대로 포켓몬 번호(숫자)가 들어오면 포켓몬 이름을 출력할 수 있어야 하므로 이름, 번호 두 변수를 매칭하고 있어야 하며 두 가지 경우에 대해 각각 출력이 가능하도록 해야 한다. 이를 위해서 dict 형태를 선택하였고 번호를 key로, 이름을 value로 해서 한 번 저장하고 반대로 이름을 key로, 번호를 value로 해서도 저장했다.