[Gold III] Title: LCA, Time: 1308 ms, Memory: 76072 KB -BaekjoonHub

Author: 051198Hz

백준/Gold/11437. LCA/LCA.swift

@@ -0,0 +1,67 @@
+final class Queue<T> {
+    private var queue = [T?]()
+    private var head = 0
+    var isEmpty: Bool {
+        queue.count - head == 0
+    }
+    func enqueue(_ element: T) {
+        queue.append(element)
+    }
+    func dequeue() -> T? {
+        if isEmpty { return nil }
+        let ret = queue[head]
+        queue[head] = nil
+        head += 1
+        return ret
+    }
+}
+
+var answer = ""
+let n = Int(readLine()!)!
+var graph = [[Int]](repeating: [Int](), count: n+1)
+var parentInfo = [Int](repeating: 0, count: n+1)
+var depthInfo = [Int](repeating: 0, count: n+1)
+
+(0..<n-1).forEach { _ in
+    let ab = readLine()!.split { $0 == " " }.map { Int(String($0))! }
+    graph[ab[0]].append(ab[1])
+    graph[ab[1]].append(ab[0])
+}
+
+var startNode = 1
+var queue = Queue<(Int, Int)>()
+queue.enqueue((startNode, 0))
+while !queue.isEmpty {
+    let (curNode, depth) = queue.dequeue()!
+    let childNodes = graph[curNode]
+    childNodes.forEach {
+        guard parentInfo[$0] == 0 && $0 != 1 else { return }
+        parentInfo[$0] = curNode
+        depthInfo[$0] = (depth+1)
+        queue.enqueue(($0, depth+1))
+    }
+}
+let m = Int(readLine()!)!
+(0..<m).forEach { _ in
+    let vg = readLine()!.split { $0 == " " }.map { Int(String($0))! }
+    // v의 depth가 g보다 반드시 큼
+    var v = vg[0]
+    var g = vg[1]
+    if depthInfo[v] < depthInfo[g] { swap(&v, &g) }
+    
+    while depthInfo[v] > depthInfo[g] {
+        v = parentInfo[v]
+    }
+    
+    if v == g {
+        answer.write("\(v)\n")
+        return
+    }
+    
+    while parentInfo[v] != parentInfo[g] {
+        v = parentInfo[v]
+        g = parentInfo[g]
+    }
+    answer.write("\(parentInfo[v])\n")
+}
+print(answer)

백준/Gold/11437. LCA/README.md

@@ -0,0 +1,30 @@
+# [Gold III] LCA - 11437 
+
+[문제 링크](https://www.acmicpc.net/problem/11437) 
+
+### 성능 요약
+
+메모리: 76072 KB, 시간: 1308 ms
+
+### 분류
+
+그래프 이론, 최소 공통 조상, 트리
+
+### 제출 일자
+
+2025년 3월 22일 22:51:41
+
+### 문제 설명
+
+<p>N(2 ≤ N ≤ 50,000)개의 정점으로 이루어진 트리가 주어진다. 트리의 각 정점은 1번부터 N번까지 번호가 매겨져 있으며, 루트는 1번이다.</p>
+
+<p>두 노드의 쌍 M(1 ≤ M ≤ 10,000)개가 주어졌을 때, 두 노드의 가장 가까운 공통 조상이 몇 번인지 출력한다.</p>
+
+### 입력 
+
+ <p>첫째 줄에 노드의 개수 N이 주어지고, 다음 N-1개 줄에는 트리 상에서 연결된 두 정점이 주어진다. 그 다음 줄에는 가장 가까운 공통 조상을 알고싶은 쌍의 개수 M이 주어지고, 다음 M개 줄에는 정점 쌍이 주어진다.</p>
+
+### 출력 
+
+ <p>M개의 줄에 차례대로 입력받은 두 정점의 가장 가까운 공통 조상을 출력한다.</p>
+