[swift] 백준 - 1764번: 듣보잡

1764번: 듣보잡

문제 푸는 데 있어 도움이 되도록 나의 풀이와 개선된 풀이를 올립니다.
또한 풀이 후 다른 사람의 풀이를 보고 참고할만한 풀이도 올립니다.
- 문제에 따라 나의 풀이만 있을 수 있습니다.
- 해당 풀이들은 풀이 중 하나일 뿐 최선의 풀이는 아닐 수 있습니다.

 

---

 

나의 풀이

let input = readLine()!.split(separator: " ").map { Int($0)! }
var arr = Set<String>()
var result = [String]()
for _ in 0..<input[0] {
    arr.update(with: readLine()!)
}
for _ in 0..<input[1] {
    let name = readLine()!
    if arr.contains(name) {
        result.append(name)
    }
}
print(result.count)
for i in result.sorted() { print(i) }

 

Set을 이용해 문제를 풀 수 있었다.

 

일반 배열에서 contains의 시간복잡도는 O(n)으로 시간초과가 나올게 뻔하기 때문에

문제는 중복되는 이름이 없기 때문에 Set의 contains를 이용해 풀었다.

Set의 contains의 시간복잡도는 O(1)이다.

---

 

 

참고할만한 풀이

코드

 

---