[백준/Swift] 10816번: 숫자 카드 2

 

https://www.acmicpc.net/problem/10816

10816번: 숫자 카드 2

 

처음에는 이분탐색으로 했는데 시간초과 나서 딕셔너리를 사용해서 해당 카드의 개수를 세고 출력해 주는 방식으로 했다.

 

 

[시간 초과 코드]

import Foundation

let n = Int(readLine()!)!
var sg = readLine()!.split(separator: " ").map{ Int(String($0))! }.sorted()
let m = Int(readLine()!)!
var card = readLine()!.split(separator: " ").map{ Int(String($0))! }
var answer = [Int]()

for i in 0..<m {
    let target = card[i]
    guard sg.contains(target) else {
        answer.append(0)
        continue
    }
    let first = sg.firstIndex(of: target)!
    let last = sg.lastIndex(of: target)!
    answer.append(last - first + 1)
}
print(answer.map{ String($0) }.joined(separator: " "))

 

[정답 코드]

import Foundation

let n = Int(readLine()!)!
var sgDict = [String: Int]()
var sg = readLine()!.split(separator: " ").map{ String($0) }
for i in sg {
    sgDict[i] = (sgDict[i] ?? 0) + 1
}
let m = Int(readLine()!)!
var card = readLine()!.split(separator: " ").map{ String($0) }
var answer = [String]()

card.forEach {
    if sgDict[$0] == nil {
        answer.append("0")
    }else {
        answer.append(String(sgDict[$0]!))
    }
}
print(answer.joined(separator: " "))