[Gold V] Title: 트리, Time: 8 ms, Memory: 79508 KB -BaekjoonHub

Author: 051198Hz

백준/Gold/1068. 트리/README.md

@@ -0,0 +1,40 @@
+# [Gold V] 트리 - 1068 
+
+[문제 링크](https://www.acmicpc.net/problem/1068) 
+
+### 성능 요약
+
+메모리: 79508 KB, 시간: 8 ms
+
+### 분류
+
+그래프 이론, 그래프 탐색, 트리, 깊이 우선 탐색
+
+### 제출 일자
+
+2025년 5월 30일 19:57:13
+
+### 문제 설명
+
+<p>트리에서 리프 노드란, 자식의 개수가 0인 노드를 말한다.</p>
+
+<p>트리가 주어졌을 때, 노드 하나를 지울 것이다. 그 때, 남은 트리에서 리프 노드의 개수를 구하는 프로그램을 작성하시오. 노드를 지우면 그 노드와 노드의 모든 자손이 트리에서 제거된다.</p>
+
+<p>예를 들어, 다음과 같은 트리가 있다고 하자.</p>
+
+<p style="text-align: center"><img alt="" src="https://upload.acmicpc.net/560de878-d961-475e-ada4-e1f0774e5a84/-/preview/" style="width: 200px; height: 185px;"></p>
+
+<p>현재 리프 노드의 개수는 3개이다. (초록색 색칠된 노드) 이때, 1번을 지우면, 다음과 같이 변한다. 검정색으로 색칠된 노드가 트리에서 제거된 노드이다.</p>
+
+<p style="text-align: center"><img alt="" src="https://upload.acmicpc.net/d46ddf4e-1b82-44cc-8c90-12f76e5bf88f/-/preview/" style="width: 200px; height: 185px;"></p>
+
+<p>이제 리프 노드의 개수는 1개이다.</p>
+
+### 입력 
+
+ <p>첫째 줄에 트리의 노드의 개수 N이 주어진다. N은 50보다 작거나 같은 자연수이다. 둘째 줄에는 0번 노드부터 N-1번 노드까지, 각 노드의 부모가 주어진다. 만약 부모가 없다면 (루트) -1이 주어진다. 셋째 줄에는 지울 노드의 번호가 주어진다.</p>
+
+### 출력 
+
+ <p>첫째 줄에 입력으로 주어진 트리에서 입력으로 주어진 노드를 지웠을 때, 리프 노드의 개수를 출력한다.</p>
+

백준/Gold/1068. 트리/트리.swift

@@ -0,0 +1,62 @@
+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()
+let parentList = (0..<n).map { _ in io.readInt() }
+let targetRemoveNode = io.readInt()
+
+var leafNodeCount = [Int](repeating: 0, count: n)
+var graph = [[Int]](repeating: [Int](), count: n)
+
+let parentRemovedNode = parentList[targetRemoveNode]
+
+//　부모노드정보로 트리를 구성
+// 트리 구성 후 리프노드를 카운트
+// 루트노드의 리프노트카운트에서 해당 노드의 리프노드 카운트를 뺀 값이 정답
+var rootNode = 0
+
+parentList.enumerated().forEach { node, parent in
+    if parent == -1 {
+        rootNode = node
+        return
+    }
+    if node == targetRemoveNode {
+        return
+    }
+    if parent == targetRemoveNode {
+        return
+    }
+    graph[parent].append(node)
+}
+
+func dfs(_ node: Int, _ targetRemoveNode: Int) -> Int {
+    if graph[node].isEmpty {
+        return 1
+    }
+    for childNode in graph[node] {
+        leafNodeCount[node] += dfs(childNode, targetRemoveNode)
+    }
+    return leafNodeCount[node]
+}
+
+let answer = targetRemoveNode == rootNode ? 0 : dfs(rootNode, targetRemoveNode)
+
+print(answer)