Closed
Description
💬 문제
[코딩테스트 연습 - 전력망을 둘로 나누기](https://school.programmers.co.kr/learn/courses/30/lessons/86971)
💬 Idea
첫번째 풀이
-
아이디어 - 간선을 가장 많이 갖고있는 노드 사이를 끊자
-
결과 → 1,6,7 만 성공 ..
-
반례 → 이 case에서 가장 많은 간선을 갖고있는 노드는 3 or 6이다.
- 3과 5 사이를 끊었을 때: 3 / 6으로 차이값은 3이 된다. (6-7 사이도 동일)
- 그러나, 이 case에서는 4와 5 사이나, 4와 6 사이를 끊는 것이 더 적은 차이값을 도출한다.
- 따라서 이 아이디어로 문제를 풀게 되면 대부분의 테스트케이스를 실패하게 된다.
두번째 풀이
- 아이디어 - 그래프를 돌면서 전력망을 하나씩 끊어보자 (완전탐색)
- 전력망을 끊은 후 - dfs로 네트워크의 수를 구한 후 차이값을 도출하고 - 전력망을 다시 연결해주는 것을 반복한다.
- 만약 6과 7사이를 끊는다면, 전력망을 끊은 후 dfs로 6, 7 각각의 수에서 출발하였을 때 송전탑의 개수를 도출하여 차이값을 구한다.
💬 풀이
첫번째 풀이
(실패)
func solution(n:Int, wires:[[Int]]) -> Int {
var v1: [Int: [Int]] = [:]
for wire in wires {
if v1[wire[0]] != nil {
v1[wire[0]]?.append(wire[1])
} else {
v1[wire[0]] = [wire[1]]
}
if v1[wire[1]] != nil {
v1[wire[1]]?.append(wire[0])
} else {
v1[wire[1]] = [wire[0]]
}
}
let maxkey = v1.sorted(by: { $0.value.count > $1.value.count }).first!.key
var cnt = (0,0)
for i in v1[maxkey]! {
cnt.0 = v1[i]!.count >= cnt.1 ? i : cnt.0
cnt.1 = v1[i]!.count >= cnt.1 ? v1[i]!.count : cnt.1
}
v1[maxkey] = v1[maxkey]?.filter({ $0 != cnt.0 })
v1[cnt.0] = v1[cnt.0]?.filter({ $0 != maxkey })
var v2: [Int: [Int]] = [:]
v2[cnt.0] = v1[cnt.0]
v1[cnt.0]?.forEach { k in
v2[k] = v1[k]
v1 = v1.filter({ $0.key != k })
}
v1 = v1.filter({ $0.key != cnt.0 })
return abs(dfs(v1, v1.first!.key).count - dfs(v2, v2.first!.key).count)
}
func dfs(_ arr: [Int: [Int]], _ n: Int) -> [Int] {
var visitedQueue: [Int] = []
var needVisitStack: [Int] = [n]
while !needVisitStack.isEmpty {
let node: Int = needVisitStack.removeLast()
if visitedQueue.contains(node) { continue }
visitedQueue.append(node)
needVisitStack += arr[node] ?? []
}
return visitedQueue
}
두번째 풀이
func solution(n:Int, wires:[[Int]]) -> Int {
var route: [Int: [Int]] = [:]
// 그래프 만들기
for wire in wires {
if route[wire[0]] != nil {
route[wire[0]]?.append(wire[1])
} else {
route[wire[0]] = [wire[1]]
}
if route[wire[1]] != nil {
route[wire[1]]?.append(wire[0])
} else {
route[wire[1]] = [wire[0]]
}
}
var result = wires.count
for i in route {
for j in i.value {
// 끊기
route[i.key] = route[i.key]!.filter({ $0 != j })
route[j] = route[j]!.filter({ $0 != i.key })
let distance = abs(dfs(route, i.key).count - dfs(route, j).count)
result = distance < result ? distance : result
// 복구
route[i.key]?.append(j)
route[j]?.append(i.key)
}
}
return result
}
func dfs(_ arr: [Int: [Int]], _ n: Int) -> [Int] {
var visitedQueue: [Int] = []
var needVisitStack: [Int] = [n]
while !needVisitStack.isEmpty {
let node: Int = needVisitStack.removeLast()
if visitedQueue.contains(node) { continue }
visitedQueue.append(node)
needVisitStack += arr[node] ?? []
}
return visitedQueue
}