백준 15681 파이썬
[골드 5] 백준 15681 - 트리와 쿼리 (파이썬)
[골드 5] 백준 15681 - 트리와 쿼리 (파이썬)
2025.04.28https://www.acmicpc.net/problem/15681풀이루트가 R인 트리가 주어짐각 쿼리는 정점 U를 루트로 하는 서브트리의 정점 개수를 묻는다여러 쿼리에 대해 빠르게 답변해야 하므로 DFS를 활용한 서브트리 크기 미리 계산이 핵심adj = [[] for _ in range(N)]adj[u]에 연결된 정점들 v를 저장하여 무방향 트리를 구성한다.D = [0] * ND[u]는 u를 루트로 하는 서브트리의 정점 수를 나타낸다.def dfs(u): visit[u] = True D[u] = 1 # 자기 자신 포함 for v in adj[u]: if not visit[v]: dfs(v) D[u] += D[v]DFS를 통한 서브트리..