[BaekJoon] 1620번 : 나는야 포켓몬 마스터 이다솜 (Python)

코딩 1일 1문제! 오늘의 문제는 백준의 나는야 포켓몬 마스터 이다솜 입니다.

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

👨🏻‍💻 코드 ( Solution )

import sys

input = sys.stdin.readline

def i_am_pocketmon_master_dasom_lee(pocketmon_dogam, quiz_list):
    dogam_pocketmons = [item[0] for item in sorted(list(pocketmon_dogam.items()), key=lambda x: x[1])]
    
    for quiz in quiz_list:
        if quiz.isdigit():
            print(dogam_pocketmons[int(quiz)-1])
        else:
            print(pocketmon_dogam[quiz])
            

if __name__ == "__main__":
    M, N = map(int, input().split())
    
    pocketmon_dogam = {}
    quiz_list = []
    
    for i in range(1, M+1):
        pocketmon = input().rstrip()
        pocketmon_dogam[pocketmon] = i
        
    for i in range(N):
        quiz = input().rstrip()
        quiz_list.append(quiz)
        
    i_am_pocketmon_master_dasom_lee(pocketmon_dogam, quiz_list)

GitHub - SOMJANG/CODINGTEST_PRACTICE: 1일 1문제 since 2020.02.07