Skip to content

Commit 04404e2

Browse files
authored
[20251117] BOJ / G5 / 노드사이의 거리 / 설진영
1 parent e42304a commit 04404e2

File tree

1 file changed

+84
-0
lines changed

1 file changed

+84
-0
lines changed
Lines changed: 84 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,84 @@
1+
```java
2+
import java.io.*;
3+
import java.util.*;
4+
5+
public class Main {
6+
static List<List<Edge>> graph;
7+
static int n, m;
8+
9+
public static void main(String[] args) throws IOException {
10+
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
11+
BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
12+
StringTokenizer st = new StringTokenizer(br.readLine());
13+
14+
n = Integer.parseInt(st.nextToken());
15+
m = Integer.parseInt(st.nextToken());
16+
17+
graph = new ArrayList<>();
18+
for (int i = 0; i <= n; i++) {
19+
graph.add(new ArrayList<>());
20+
}
21+
22+
for (int i = 0; i < n - 1; i++) {
23+
st = new StringTokenizer(br.readLine());
24+
int u = Integer.parseInt(st.nextToken());
25+
int v = Integer.parseInt(st.nextToken());
26+
int w = Integer.parseInt(st.nextToken());
27+
28+
graph.get(u).add(new Edge(v, w));
29+
graph.get(v).add(new Edge(u, w));
30+
}
31+
32+
for (int i = 0; i < m; i++) {
33+
st = new StringTokenizer(br.readLine());
34+
int start = Integer.parseInt(st.nextToken());
35+
int end = Integer.parseInt(st.nextToken());
36+
37+
int distance = bfs(start, end);
38+
bw.write(distance + "\n");
39+
}
40+
41+
bw.flush();
42+
bw.close();
43+
br.close();
44+
}
45+
46+
static int bfs(int start, int end) {
47+
ArrayDeque<Integer> queue = new ArrayDeque<>();
48+
int[] dist = new int[n + 1];
49+
boolean[] visited = new boolean[n + 1];
50+
51+
Arrays.fill(dist, -1);
52+
queue.offer(start);
53+
dist[start] = 0;
54+
visited[start] = true;
55+
56+
while (!queue.isEmpty()) {
57+
int curr = queue.poll();
58+
59+
if (curr == end) {
60+
return dist[end];
61+
}
62+
63+
for (Edge edge : graph.get(curr)) {
64+
if (!visited[edge.to]) {
65+
visited[edge.to] = true;
66+
dist[edge.to] = dist[curr] + edge.weight;
67+
queue.offer(edge.to);
68+
}
69+
}
70+
}
71+
72+
return dist[end];
73+
}
74+
75+
static class Edge {
76+
int to, weight;
77+
78+
Edge(int to, int weight) {
79+
this.to = to;
80+
this.weight = weight;
81+
}
82+
}
83+
}
84+
```

0 commit comments

Comments
 (0)