-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathBOJ1504.java
More file actions
109 lines (89 loc) · 1.97 KB
/
BOJ1504.java
File metadata and controls
109 lines (89 loc) · 1.97 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
package quki.algorithm.dijkstra;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.Scanner;
/**
*
* 다익스트라
* 특정 경로를 지나야함
* 시작점을 다르게 해서 구함.
*
* @author quki
*
*/
public class BOJ1504 {
public static final int inf = 100000000; // 한자리 낮춤. inf+inf+inf일 수 있어서
public static class Edge {
int v;
int c;
Edge(int v, int c) {
this.v = v;
this.c = c;
}
}
public static int dijkstra(int start, int end, HashMap<Integer, ArrayList<Edge>> hm) {
int n = hm.size() - 1;
int d[] = new int[n + 1];
boolean check[] = new boolean[n + 1];
Arrays.fill(d, inf);
d[start] = 0;
for (int i = 1; i <= n - 1; i++) {
check[start] = true;
// Relaxation
for (Edge e : hm.get(start)) {
int u = start;
int v = e.v;
int c = e.c;
if (d[v] > d[u] + c) {
d[v] = d[u] + c;
}
}
// 다음 start 찾기
int min = inf;
for (int j = 1; j <= n; j++) {
if (d[j] < min && !check[j]) {
min = d[j];
start = j;
}
}
}
return d[end];
}
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int n = sc.nextInt();
int m = sc.nextInt();
HashMap<Integer, ArrayList<Edge>> hm = new HashMap<>();
for (int i = 0; i <= n; i++) {
hm.put(i, new ArrayList<Edge>());
}
while (m-- > 0) {
int u = sc.nextInt();
int v = sc.nextInt();
int c = sc.nextInt();
hm.get(u).add(new Edge(v, c));
hm.get(v).add(new Edge(u, c));
}
int v1 = sc.nextInt();
int v2 = sc.nextInt();
int start = 1;
int end = n;
// 두가지 경우가 존재
// 1 - v1 - v2 - n
// 1 - v2 - v1 - n
int p1 = dijkstra(start, v1, hm);
int p2 = dijkstra(v1, v2, hm);
int p3 = dijkstra(v2, end, hm);
int ans1 = p1 + p2 + p3;
p1 = dijkstra(start, v2, hm);
p2 = dijkstra(v2, v1, hm);
p3 = dijkstra(v1, end, hm);
int ans2 = p1 + p2 + p3;
int ans = Math.min(ans1, ans2);
if (ans >= inf) {
ans = -1;
}
System.out.println(ans);
}
}