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

문제 링크

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

문제 풀이

나의 풀이

  

    
    
    
    
  
import Foundation

let NM = readLine()!.split(separator: " " ).map{Int(String($0))!}

var array1 = [String : Int]()
var array2 = [Int : String]()

for i in 1...NM[0] {
    let p = readLine()!
    array1[p] = i
    array2[i] = p
}

for _ in 1...NM[1] {
    let find = readLine()!
    
    if let poketmon = Int(find) {
        print(array2[poketmon]!)
    } else {
        print(array1[find]!)
    }
}

풀이 과정

  

    
    
    
    
  
// 시간초과 코드

import Foundation

let NM = readLine()!.split(separator: " " ).map{Int(String($0))!}

var array : [String] = []

for _ in 1...NM[0] {
    let p = readLine()!
    array.append(p)
}

for _ in 1...NM[1] {
    let find = readLine()!
    if Int(find) != nil {
        print(array[Int(find)!-1])
    } else {
        print((array.firstIndex(of: find))!+1)
    }
}

Swift... 시간초과가 많이 나는 것 같다. 특히 배열을 사용할 때 append, firstIndex를 써서 그런가 시간초과가 났다. 그래서 배열을 최대한 빼줬다.

  

    
    
    
    
  
import Foundation

let NM = readLine()!.split(separator: " " ).map{Int(String($0))!}

var array1 = [String : Int]()
var array2 = [Int : String]()

for i in 1...NM[0] {
    let p = readLine()!
    array1[p] = i
    array2[i] = p
}

for _ in 1...NM[1] {
    let find = readLine()!
    
    if let poketmon = Int(find) {
        print(array2[poketmon]!)
    } else {
        print(array1[find]!)
    }
}


// 아래 코드도 가능하다!

// for _ in 1...NM[1] {
//     let find = readLine()!
// 
//     if Int(find) != nil {
//         print(array2[Int(find)!]!)
//     } else {
//         print(array1[find]!)
//     }
// }

딕셔너리를 사용했다. if let도 이용해 보았다. poketmon이 정의되면 array2에서 가져오고, nil값이 되면 array1에서 가져온다!

 

!를 붙여서 Optional을 강제로 벗겨줬는데 이 방법이 좋은 것인지 모르겠다. 좀 더 공부가 필요하다.