[Swift] 백준2630 나는야 포켓몬 마스터 이다솜

import Foundation
    
let NM = readLine()!.components(separatedBy: " ").map { Int($0)! }

let N = NM[0]
let M = NM[1]

var pokedex1 = [String: Int]()
var pokedex2 = [Int: String]()

for i in 0..<N{
    let pokmon = readLine()!
    pokedex1[pokmon] = i+1
    pokedex2[i+1] = pokmon
}

// 문제가 알파벳으로만 들어오면 포켓몬 번호를 말해야 하고
// 숫자로만 들어오면 포켓몬 번호에 해당하는 문자를 출력해야함
for _ in 0..<M{
    let q = readLine()!
    if let index = Int(q) {
        print(pokedex2[index]!)
    } else {
        print(pokedex1[q]!)
    }
}

시간초과가 나와서 딕셔너리 2개 만들어서 풀었다.