DFS / 잊지말자 트리에서도 DFS BFS

Author: SoTaeHo

15681번 - 트리와 쿼리/Main.java

@@ -0,0 +1,66 @@
+
+/* ************************************************************************** */
+/*                                                                            */
+/*                                                      :::    :::    :::     */
+/*   Problem Number: 15681                             :+:    :+:      :+:    */
+/*                                                    +:+    +:+        +:+   */
+/*   By: thxogh1 <boj.kr/u/thxogh1>                  +#+    +#+          +#+  */
+/*                                                  +#+      +#+        +#+   */
+/*   https://boj.kr/15681                          #+#        #+#      #+#    */
+/*   Solved: 2025/05/05 19:14:42 by thxogh1       ###          ###   ##.kr    */
+/*                                                                            */
+/* ************************************************************************** */
+import java.io.BufferedReader;
+import java.io.InputStreamReader;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.StringTokenizer;
+
+public class Main {
+    static List<List<Integer>> li;
+    static int[] ans;
+    static boolean[] visited;
+
+    static int dfs(int node) {
+        if (li.get(node).size() == 0) {
+            ans[node] = 1;
+            return 1;
+        }
+        if (visited[node]) {
+            return 0;
+        }
+        visited[node] = true;
+        int cnt = 0;
+        for (int i = 0; i < li.get(node).size(); i++) {
+            cnt += dfs(li.get(node).get(i));
+        }
+        ans[node] = cnt + 1;
+        return cnt + 1;
+    }
+
+    public static void main(String[] args) throws Exception {
+        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
+        StringTokenizer st = new StringTokenizer(br.readLine());
+
+        int n = Integer.parseInt(st.nextToken());
+        int r = Integer.parseInt(st.nextToken());
+        int q = Integer.parseInt(st.nextToken());
+        ans = new int[n + 1];
+        li = new ArrayList<>();
+        for (int i = 0; i < n + 1; i++) {
+            li.add(new ArrayList<>());
+        }
+        for (int i = 0; i < n - 1; i++) {
+            st = new StringTokenizer(br.readLine());
+            int s = Integer.parseInt(st.nextToken());
+            int e = Integer.parseInt(st.nextToken());
+            li.get(s).add(e);
+            li.get(e).add(s);
+        }
+        visited = new boolean[n + 1];
+        dfs(r);
+        for (int i = 0; i < q; i++) {
+            System.out.println(ans[Integer.parseInt(br.readLine())]);
+        }
+    }
+}

15681번 - 트리와 쿼리/README.md

@@ -0,0 +1,39 @@
+# 15681번: 트리와 쿼리 - <img src="https://static.solved.ac/tier_small/11.svg" style="height:20px" /> Gold V
+
+<!-- performance -->
+
+<!-- 문제 제출 후 깃허브에 푸시를 했을 때 제출한 코드의 성능이 입력될 공간입니다.-->
+
+<!-- end -->
+
+## 문제
+
+[문제 링크](https://boj.kr/15681)
+
+<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의 형태로 트리에 속한&nbsp;간선의 정보가 주어진다. (1 ≤ U, V ≤ N, U ≠ V)</p>
+
+<p>이는 U와 V를 양 끝점으로 하는 간선이 트리에 속함을 의미한다.</p>
+
+<p>이어 Q줄에 걸쳐, 문제에 설명한 U가 하나씩 주어진다. (1 ≤ U ≤ N)</p>
+
+<p>입력으로 주어지는 트리는 항상 올바른 트리임이 보장된다.</p>
+
+## 출력
+
+<p>Q줄에 걸쳐 각 쿼리의 답을 정수 하나로 출력한다.</p>
+
+## 소스코드
+
+[소스코드 보기](Main.java)