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

📍 백준 1620 - 나는야 포켓몬 마스터 이다솜

백준 1620 - 나는야 포켓몬 마스터 이다솜

---

⚡️ 나의 풀이

포켓몬을 좋아해서 풀어봤다.

 

입력의 범위가 100,000까지로 주어져있기 때문에 리스트로 풀기에 시간 초과가 날 것 같아 dict의 hash를 이용하여 풀었다. (hash 접근 시간복잡도는 O(1))

 

dict에서 value의 값을 찾는것은 dict[key]를 넣으면 되기 때문에 비교적 쉬웠으나 반대로 key를 찾는것은 dict로는 찾을 수 없기 때문에 dict.keys()만을 뽑아 리스트로 만드는 list(dict.keys())를 사용했고, 여기서 몇 번째인지 알기 위해 인덱싱을 사용했다.

 

1. 입력을 key:value 형태인 dict로 넣어준다.(이때, key는 포켓몬 이름, value는 입력 순서)
2. key만 모아 새로운 list를 생성한다.
3. 입력이 isdigit()면 key를 출력하고 not isdigit()면 value를 출력한다.

 

# 나의 코드
import sys
input = sys.stdin.readline

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

for i in range(1, n+1):
    name = input().strip()
    pocket_books[name] = i

pocket_books_keys = list(pocket_books.keys())

for _ in range(m):
    name_or_number = input().strip()
    if name_or_number.isdigit():
        print(pocket_books_keys[int(name_or_number) - 1])
    else:
        print(pocket_books[name_or_number])