[Swift 알고리즘] 백준 1764 듣보잡

○

정렬 - Silver 4 (30분)

 

백준 1764 듣보잡

문제
김진영이 듣도 못한 사람의 명단과, 보도 못한 사람의 명단이 주어질 때, 듣도 보도 못한 사람의 명단을 구하는 프로그램을 작성하시오.
입력
첫째 줄에 듣도 못한 사람의 수 N, 보도 못한 사람의 수 M이 주어진다. 이어서 둘째 줄부터 N개의 줄에 걸쳐 듣도 못한 사람의 이름과, N+2째 줄부터 보도 못한 사람의 이름이 순서대로 주어진다. 이름은 띄어쓰기 없이 알파벳 소문자로만 이루어지며, 그 길이는 20 이하이다. N, M은 500,000 이하의 자연수이다.
듣도 못한 사람의 명단에는 중복되는 이름이 없으며, 보도 못한 사람의 명단도 마찬가지이다.
출력
듣보잡의 수와 그 명단을 사전순으로 출력한다.

 

첫번째 시도 - 시간초과

당연히 시간초과 뜰 걸 알았지만 일단 단순하게 먼저 구현해봄

1. N 수만큼 입력값을 받아오고

2. M 수만큼 입력값 받아오면서 하나씩 n.contains(name) 체크하고 결과 명단 배열에 넣어주기

3. 결과 명단 배열 반복문 돌리며 print

contains 가 O(n)을 시간복잡도로 가지는 메서드이기 때문에 두번째 과정에서 O(n^2) 최대 50만 x 50만으로 시간초과가 뜰 수 밖에..

for i in 0..<nm[1] {
    let name = readLine()!
    if n.contains(name) {
        result.append(name)
    }
}

 

두번째 시도 - 정렬 79244KB 68ms

그렇다면 음..

어짜피 출력을 사전순으로 하니까 순서대로 정렬을 먼저 하고 중복되는 이름을 찾아야 겠다고 생각함.

1. N + M 수만큼 입력값을 동시에 받아오고 (어짜피 중복만 체크하면 되니까 구분할 필요 업슴)

2. 받아온 명단 배열을 정렬해준 후

3. 반복문으로 명단 배열에서 중복되는 이름을 찾아 결과 명단 배열에 넣어주기

4. 결과 명단 배열 반복문 돌리며 print

메모리적으로나 시간적으로나 꽤 나쁘지 않은 풀이법인 것 같음.

let nm = readLine()!.split(separator: " ").map { Int($0)! }
var names = [String](repeating: "", count: nm[0] + nm[1])
var result = [String]()
// 1
for i in 0..<(nm[0] + nm[1]) {
    names[i] = readLine()!
}
// 2
names.sort()
// 3
for i in 1..<names.count {
    if names[i-1] == names[i] {
        result.append(names[i])
    }
}
// 4
print(result.count)
for name in result {
    print(name)
}

 

이거 말고도 딕셔너리에 저장하는 방법도 있고

뭔가 중복을 찾는거면 Set을 사용해서 풀 수 있는 방법이 있지 않을까 궁금했는데 

Set 을 사용한 풀이법이 있어 가져와보았다. 

 

Set 을 사용한 풀이 - 75796KB 72ms

첫번째 시도에서랑 풀이법은 동일한데, 배열 대신 Set 을 사용하면 contains 메서드의 시간복잡도가 O(1) 이기 때문에 훨씬 이득이다!

let nm = readLine()!.split(separator: " ").map { Int($0)! }
let n = Set((0..<nm[0]).map { _ in readLine()! })
var result = Set<String>()

for _ in 0..<nm[1] {
    let name = readLine()!
    if n.contains(name) { result.insert(name) }
}

print(result.count)
for name in result.sorted() { print(name) }

 

앞으로 Set 을 애용하도록 해보자..!