백준 1620번 포켓몬 마스터

사실 너무 쉬운문제라 포스팅할 생각은 없었는데

 

입력이 도감에 있는 문자가 아니라 전부 소문자 혹은 전부 대문자인 경우도 들어오는줄알아서 소문자 대문자 변환 라이브러리 안쓰고 풀고싶어서 어떻게 풀지 생각을 너무 오래했는데 그런 문제가 아니었다...

 

그래서 그냥 이렇게 된거 포스팅까지 하기로 했다.

 

import sys

n,m=map(int,sys.stdin.readline().rstrip().split())
arr=[]
board={}
for i in range(n):
    arr.append(sys.stdin.readline().rstrip())
    board[arr[i]]=i+1

for i in range(m):
    j=sys.stdin.readline().rstrip()
    if ord(j[0])>ord('0') and ord(j[0])<=ord('9'):
        print(arr[int(j)-1])
        continue
    print(board[j])

제출하자