백준 1620번 나는야 포켓몬 마스터 이다솜 파이썬 풀이 시간 초과 해결

N,M=map(int,input().split())
dic={}
for i in range (1, N+1):
    temp=input()
    dic[i]=temp
    dic[temp]=i
    
    
for k in range (M):
    temp=input()
    if( temp.isdigit()):
        print(dic[int(temp)])
    else:
        print(dic[temp])

 

문제가 살짝 병맛..? 포켓몬을 재밌게 하고 만들었나 (｡･ˇ_ˇ･｡)

 

처음에는 역시 런타임 에러 

for문으로 무지성 돌리기 했는데 시간 초과가 난다 

 

정답은 딕셔너리 형태로 저장하는것! 

{1:피카츄, 피카츄:1, 2:꼬부기, 꼬부기:2  ........}

이런식으로 저장한다 

 

문자가 정수인지 확인하는 함수

temp.isdigit() 

변수뒤에 isdigit()을 붙여서 정수이면 True가 나온다 

 

그리고 분명 맞는 풀이인데 정답인정이 안되어서 

설정을 Python3에서 PyPy3으로 바꾸니 정답처리가 되더라..

혹시 파이썬에서 분명 맞는데 정답처리가 안 되는 사람들은 언어를 바꾸어 보세요! 

 

 

#백준 #코딩 #코딩공부