[Gold V] Title: 트리와 쿼리, Time: 84 ms, Memory: 103508 KB -BaekjoonHub

Author: 051198Hz

백준/Gold/15681. 트리와 쿼리/README.md

@@ -0,0 +1,42 @@
+# [Gold V] 트리와 쿼리 - 15681 
+
+[문제 링크](https://www.acmicpc.net/problem/15681) 
+
+### 성능 요약
+
+메모리: 103508 KB, 시간: 84 ms
+
+### 분류
+
+다이나믹 프로그래밍, 그래프 이론, 그래프 탐색, 트리, 깊이 우선 탐색, 트리에서의 다이나믹 프로그래밍
+
+### 제출 일자
+
+2025년 5월 30일 18:48:14
+
+### 문제 설명
+
+<p>간선에 가중치와 방향성이 없는 임의의 루트 있는 트리가 주어졌을 때, 아래의 쿼리에 답해보도록 하자.</p>
+
+<ul>
+	<li>정점 U를 루트로 하는 서브트리에 속한 정점의 수를 출력한다.</li>
+</ul>
+
+<p>만약 이 문제를 해결하는 데에 어려움이 있다면, 하단의 힌트에 첨부한 문서를 참고하자.</p>
+
+### 입력 
+
+ <p>트리의 정점의 수 N과 루트의 번호 R, 쿼리의 수 Q가 주어진다. (2 ≤ N ≤ 10<sup>5</sup>, 1 ≤ R ≤ N, 1 ≤ Q ≤ 10<sup>5</sup>)</p>
+
+<p>이어 N-1줄에 걸쳐, U V의 형태로 트리에 속한 간선의 정보가 주어진다. (1 ≤ U, V ≤ N, U ≠ V)</p>
+
+<p>이는 U와 V를 양 끝점으로 하는 간선이 트리에 속함을 의미한다.</p>
+
+<p>이어 Q줄에 걸쳐, 문제에 설명한 U가 하나씩 주어진다. (1 ≤ U ≤ N)</p>
+
+<p>입력으로 주어지는 트리는 항상 올바른 트리임이 보장된다.</p>
+
+### 출력 
+
+ <p>Q줄에 걸쳐 각 쿼리의 답을 정수 하나로 출력한다.</p>
+

백준/Gold/15681. 트리와 쿼리/트리와 쿼리.swift

@@ -0,0 +1,54 @@
+import Foundation
+
+class FileIO {
+    @inline(__always) private var buffer: [UInt8] = Array(FileHandle.standardInput.readDataToEndOfFile()) + [0], byteIdx = 0
+    
+    @inline(__always) private func readByte() -> UInt8 {
+        defer { byteIdx += 1 }
+        return buffer.withUnsafeBufferPointer { $0[byteIdx] }
+    }
+    
+    @inline(__always) func readInt() -> Int {
+        var number = 0, byte = readByte(), isNegative = false
+        while byte == 10 || byte == 32 { byte = readByte() }
+        if byte == 45 { byte = readByte(); isNegative = true }
+        while 48...57 ~= byte { number = number * 10 + Int(byte - 48); byte = readByte() }
+        return number * (isNegative ? -1 : 1)
+    }
+}
+
+let io = FileIO()
+
+let n = io.readInt(),
+    r = io.readInt(),
+    q = io.readInt()
+
+var graph = [[Int]](repeating: [Int](), count: n+1)
+var subTreeSizeList = [Int](repeating: 0, count: n+1)
+
+(0..<n-1).forEach { _ in
+    let u = io.readInt()
+    let v = io.readInt()
+    graph[u].append(v)
+    graph[v].append(u)
+}
+
+func dfs(_ root: Int) {
+    subTreeSizeList[root] = 1
+    for node in graph[root] {
+        guard subTreeSizeList[node] == 0 else { continue }
+        dfs(node)
+        subTreeSizeList[root] += subTreeSizeList[node]
+    }
+}
+
+let root = r
+dfs(root)
+
+let answer = (0..<q).map { _ in
+    let query = io.readInt()
+    let answer = subTreeSizeList[query]
+    return "\(answer)\n"
+}.joined()
+
+print(answer)