문제
알고리즘
- 포켓몬의 알파벳 문제는 딕셔너리 자료형으로 수행하고 숫자 문제는 리스트 자료형으로 수행한다.
코드
import sys
n, m = map(int, sys.stdin.readline().split())
book_dic = {}
book_list = []
# 포켓몬의 개수를 반복하여 포켓몬의 이름을 입력받는다.
for i in range(n):
poketmon = sys.stdin.readline().rstrip() # 포켓몬 이름
# 포켓몬의 이름을 딕셔너리 형과 리스트 형으로 저장한다.
book_dic[poketmon] = i + 1
book_list.append(poketmon)
# 문제의 개수만큼 반복한다.
for _ in range(m):
q = sys.stdin.readline().rstrip() # 포켓몬의 알파벳이나 번호
# 입력받은 문제가 알파벳인지 숫자인지 구분한다.
# 숫자라면 리스트에 (해당 숫자 - 1) 을 입력하여 출력한다.
if q.isdigit():
print(book_list[int(q) - 1])
# 알파벳이라면 딕셔너리에 해당 알파벳을 입력하여 벨류값을 출력한다.
else:
print(book_dic[q])
github
'CodingTest > Baekjoon' 카테고리의 다른 글
[baekjoon] 백준 9375번(파이썬): 패션왕 신해빈 (0) | 2021.10.03 |
---|---|
[baekjoon] 백준 2504번(파이썬): 괄호의 값 (0) | 2021.10.02 |
[baekjoon] 백준 2493번(파이썬): 탑 (0) | 2021.09.30 |
[baekjoon] 백준 11286번(파이썬): 절댓값 힙 (0) | 2021.09.29 |
[baekjoon] 백준 5430번(파이썬): AC (0) | 2021.09.28 |