[Silver II] Title: 트리의 부모 찾기, Time: 204 ms, Memory: 125432 KB -BaekjoonHub

Author: wlsgur11

백준/Silver/11725. 트리의 부모 찾기/README.md

@@ -4,15 +4,15 @@
 
 ### 성능 요약
 
-메모리: 125496 KB, 시간: 236 ms
+메모리: 125432 KB, 시간: 204 ms
 
 ### 분류
 
 그래프 이론, 그래프 탐색, 트리, 너비 우선 탐색, 깊이 우선 탐색
 
 ### 제출 일자
 
-2025년 4월 3일 10:20:13
+2025년 6월 18일 13:05:30
 
 ### 문제 설명
 

백준/Silver/11725. 트리의 부모 찾기/트리의 부모 찾기.py

@@ -2,23 +2,28 @@
 import sys
 input = sys.stdin.readline
 
-N = int(input())
+n = int(input())
+graph = [[] for _ in range(n+1)]
+parent = [0] * (n+1)
 
-adj = [[] for _ in range(N+1)]
-for _ in range(N-1):
-    A, B = map(int, input().split())
-    adj[A].append(B)
-    adj[B].append(A)
+for _ in range(n-1):
+    a, b = map(int, input().split())
+    graph[a].append(b)
+    graph[b].append(a)
 
-parent = [0] * (N+1)
+def bfs():
+    q = deque([1])
+    parent[1] = 1
+    
+    while q:
+        now = q.popleft()
+        for next in graph[now]:
+            if parent[next] == 0:
+                parent[next] = now
+                q.append(next)
 
-q = deque([1])
-while q:
-    node = q.popleft()
-    for i in adj[node]:
-        if parent[i] == 0:
-            parent[i] = node
-            q.append(i)
+bfs()
 
-for i in range(2, N+1):
-    print(parent[i])
+for i in range(2, n+1):
+    print(parent[i])
+