[BOJ][Python] 1620 나는야 포켓몬 마스터 이다솜

[문제]

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

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

 

[코드]

아이디어: key를 숫자로 입력받고, key와 value가 반전된 딕셔너리를 만들어주어 key로 value를 찾아 출력해준다. 

- 딕셔너리를 두 개를 만드므로 메모리가 두 배로 소모된다. 

import sys
input = sys.stdin.readline

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

po = {}
for i in range(1,n+1):
    po[str(i)] = input().strip()

po_rev = {v:k for k, v in po.items()}

for i in range(m):
    q = input().strip()
    
    if q.isdigit():
        print(po.get(q))
    else:
        print(po_rev.get(q))

 

[배운점]

input().strip()

- 문자열 및 공백 제거 
- 선행과 후행 문자가 제거된 문자열의 복사본을 반환

>> 입력 시 개행문자(\n)을 제거 해줌으로써 정확한 입력값이 딕셔너리에 들어감 

 

딕셔너리명.items()

- 딕셔너리의 value를 가져온다. 

 

딕셔너리명.get(x)

- 키 값(x)로 딕셔너리의 values 즉 item을 가져온다.