[Gold II] Title: 트리의 높이와 너비, Time: 24 ms, Memory: 69908 KB -BaekjoonHub

Author: 051198Hz

백준/Gold/2250. 트리의 높이와 너비/README.md

@@ -0,0 +1,45 @@
+# [Gold II] 트리의 높이와 너비 - 2250 
+
+[문제 링크](https://www.acmicpc.net/problem/2250) 
+
+### 성능 요약
+
+메모리: 69908 KB, 시간: 24 ms
+
+### 분류
+
+그래프 이론, 그래프 탐색, 트리, 깊이 우선 탐색
+
+### 제출 일자
+
+2025년 10월 20일 15:26:45
+
+### 문제 설명
+
+<p>이진트리를 다음의 규칙에 따라 행과 열에 번호가 붙어있는 격자 모양의 틀 속에 그리려고 한다. 이때 다음의 규칙에 따라 그리려고 한다.</p>
+
+<ol>
+	<li>이진트리에서 같은 레벨(level)에 있는 노드는 같은 행에 위치한다.</li>
+	<li>한 열에는 한 노드만 존재한다.</li>
+	<li>임의의 노드의 왼쪽 부트리(left subtree)에 있는 노드들은 해당 노드보다 왼쪽의 열에 위치하고, 오른쪽 부트리(right subtree)에 있는 노드들은 해당 노드보다 오른쪽의 열에 위치한다.</li>
+	<li>노드가 배치된 가장 왼쪽 열과 오른쪽 열 사이엔 아무 노드도 없이 비어있는 열은 없다.</li>
+</ol>
+
+<p>이와 같은 규칙에 따라 이진트리를 그릴 때 각 레벨의 너비는 그 레벨에 할당된 노드 중 가장 오른쪽에 위치한 노드의 열 번호에서 가장 왼쪽에 위치한 노드의 열 번호를 뺀 값 더하기 1로 정의한다. 트리의 레벨은 가장 위쪽에 있는 루트 노드가 1이고 아래로 1씩 증가한다.</p>
+
+<p>아래 그림은 어떤 이진트리를 위의 규칙에 따라 그려 본 것이다. 첫 번째 레벨의 너비는 1, 두 번째 레벨의 너비는 13, 3번째, 4번째 레벨의 너비는 각각 18이고, 5번째 레벨의 너비는 13이며, 그리고 6번째 레벨의 너비는 12이다.</p>
+
+<p style="text-align: center;"><img alt="" src="https://upload.acmicpc.net/4e4aaa17-dc1d-4af9-a36a-3144259fb7d2/-/preview/" style="width: 488px; height: 176px;"></p>
+
+<p>우리는 주어진 이진트리를 위의 규칙에 따라 그릴 때에 너비가 가장 넓은 레벨과 그 레벨의 너비를 계산하려고 한다. 위의 그림의 예에서 너비가 가장 넓은 레벨은 3번째와 4번째로 그 너비는 18이다. 너비가 가장 넓은 레벨이 두 개 이상 있을 때는 번호가 작은 레벨을 답으로 한다. 그러므로 이 예에 대한 답은 레벨은 3이고, 너비는 18이다.</p>
+
+<p>임의의 이진트리가 입력으로 주어질 때 너비가 가장 넓은 레벨과 그 레벨의 너비를 출력하는 프로그램을 작성하시오</p>
+
+### 입력 
+
+ <p>첫째 줄에 노드의 개수를 나타내는 정수 N(1 ≤ N ≤ 10,000)이 주어진다. 다음 N개의 줄에는 각 줄마다 노드 번호와 해당 노드의 왼쪽 자식 노드와 오른쪽 자식 노드의 번호가 순서대로 주어진다. 노드들의 번호는 1부터 N까지이며, 자식이 없는 경우에는 자식 노드의 번호에 -1이 주어진다.</p>
+
+### 출력 
+
+ <p>첫째 줄에 너비가 가장 넓은 레벨과 그 레벨의 너비를 순서대로 출력한다. 너비가 가장 넓은 레벨이 두 개 이상 있을 때에는 번호가 작은 레벨을 출력한다.</p>
+

백준/Gold/2250. 트리의 높이와 너비/트리의 높이와 너비.swift

@@ -0,0 +1,73 @@
+let n = Int(readLine()!)!
+var child = [Bool](repeating: false, count: n+1)
+var tree = (0...n).map { i in Node(v: i) }
+var levels = [Int](repeating: 0, count: n)
+var maxColLevel = [Int:Int]()
+var minColLevel = [Int:Int]()
+
+for _ in 0..<n {
+    let glr = readLine()!.split { $0 == " " }.map { Int(String($0))! },
+        g = glr[0],
+        l = glr[1],
+        r = glr[2]
+    if l != -1 {
+        tree[g].left = l
+        child[l] = true
+    }
+    if r != -1 {
+        tree[g].right = r
+        child[r] = true
+    }
+}
+
+let rootNode = (1...n).filter { child[$0] == false }.first!
+
+var colPos = -1
+
+dfs(rootNode, 1)
+
+var level = 0
+var width = 0
+
+for key in maxColLevel.keys {
+    let levelWidth = maxColLevel[key]! - minColLevel[key]! + 1
+    if levelWidth == width, level > key {
+        level = key
+        width = levelWidth
+        continue
+    }
+    if levelWidth > width {
+        level = key
+        width = levelWidth
+    }
+}
+
+print(level, width)
+
+// 깊이우선탐색으로 레벨별 너비를 구한다.
+func dfs(_ current: Int, _ level: Int) {
+    if let left = tree[current].left {
+        dfs(left, level + 1)
+    }
+    
+    colPos += 1
+    levels[colPos] = level
+    
+    maxColLevel[level, default: Int.min] = colPos > maxColLevel[level, default: Int.min] ? colPos : maxColLevel[level, default: Int.min]
+    minColLevel[level, default: Int.max] = colPos < minColLevel[level, default: Int.max] ? colPos : minColLevel[level, default: Int.max]
+    
+    if let right = tree[current].right {
+        dfs(right, level + 1)
+    }
+}
+
+class Node {
+    init(v: Int, left: Int? = nil, right: Int? = nil) {
+        self.v = v
+        self.left = left
+        self.right = right
+    }
+    let v: Int
+    var left: Int?
+    var right: Int?
+}