(Python/🥈4)백준 알고리즘 1620번: 나는야 포켓몬 마스터 이다솜

문제 출처:https://www.acmicpc.net/step/49

집합과 맵 단계

풀이:

(N,M) = sys.stdin.readline().split() # 도감에 있는 포켓몬 갯수 N 문제갯수

dict = {}
# 문제가 알파벳으로만 들어오면 포켓몬 번호를 말해야 하고,
# 숫자로만 들어오면, 포켓몬 번호에 해당하는 문자를 출력해야해.
for i in range(1,int(N)+1):
    word = sys.stdin.readline().rstrip()
    dict[i] = word
    dict[word] = i                  # dict 을 만들어줌 몬스터가 키일때 숫자가 키일때 둘다

for i in range(int(M)):
    check_word = sys.stdin.readline().rstrip()
    if check_word.isdigit():               # 숫자인지 확인하는 메소드
        print(dict[int(check_word)])       # 숫자면 영어 출력하고
    else:
        print(dict[check_word])            # 영어면 숫자를 출력해라

 

이 문제는 dictionary 를 사용해서 키값과 밸류값을 지정해 둔다

 

dic에 isdigit 는 숫자인지 확인 하는 메소드다 

숫자이면 True 를 반환한다.