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

Minsu Han·2022년 9월 19일
0

알고리즘연습

목록 보기
16/105

코드

import sys
input = sys.stdin.readline

N, M = map(int, input().split())

l = [0]    # 리스트: l[번호] = 이름
d = dict()    # 딕셔너리: key=이름, value=번호

for i in range(N):
    name = input().rstrip()
    l.append(name)
    d[name] = i+1

for _ in range(M):
    q = input().rstrip()
    if q.isnumeric():
        print(l[int(q)])
    else:
        print(d[q])

결과

image


풀이 방법

  • 주어진 이름에 대한 번호를 출력하기엔 해시테이블 구조인 딕셔너리를, 번호에 대한 이름을 출력하려면 인덱스를 사용하는 리스트를 활용하면 된다고 판단하였다.
  • 주어진 입력이 숫자인지 확인하는 메소드: str.isnumberic(), str.isdigit()...

profile
기록하기

0개의 댓글