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

02 Jan 2023 in Study on Algorithm

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

문제

예제 입력

---

코드

import sys

lst1 = dict()
lst2 = list()
n, m = map(int, input().split())

for i in range(n):
    s = str(sys.stdin.readline().replace('\n', ''))
    lst1[s] = i + 1
    lst2.append(s)

for i in range(m):
    s = sys.stdin.readline().replace('\n', '')
    try :
        print(lst1[s])
    except :
        print(lst2[int(s)-1])

설명

파이썬을 통해서 사용자로부터 입력받아 반복문을 사용하여 나는야 포켓몬 마스터 이다솜을 구현했습니다.

---

정리

import sys

lst = dict()
n, m = map(int, input().rsplit())

for i in range(n):
    s = str(sys.stdin.readline().replace('\n', ''))
    lst[s] = i + 1
    lst[str(i + 1)] = s

for i in range(m):
    s = sys.stdin.readline().replace('\n', '')
    # s.isdigit()으로 문자 숫자 판별
    print(lst[s])

그 외에도 자료형 dict를 하나만 사용하여 구현 가능하며 추가적으로 isdigit()함수로 숫자 조건을 판별할 수 있습니다.

---

결과