백준 1956번
https://www.acmicpc.net/problem/1956
for (int i = 1; i <= N; i++) {
for (int j = 1; j <= N; j++) {
for (int k = 1; k <= N; k++) {
if (arr[i][j] > arr[i][k] + arr[k][j]) {
arr[i][j] = arr[i][k] + arr[k][j];
}
}
}
}
플로이드 워셜을 통해서 최단 경로를 모두 찾는다.
int ans = INF;
for (int i = 1; i <= N; i++) {
for (int j = 1; j <= N; j++) {
if (i == j) continue;
if (arr[i][j] != INF && arr[j][i] != INF) {
ans = Math.min(ans, arr[i][j] + arr[j][i]);
}
}
}
이후에는 i
== j
를 통해 자신은 제외하고
i
-> j
, j
-> i
가 서로 통행할 수 있으면 사이클이라고 판단하고 최단 거리를 구해서 답을 구할 수 있다.
import java.io.*;
import java.util.Arrays;
import java.util.StringTokenizer;
public class Main {
// input
private static BufferedReader br;
// variables
private static final int INF = Integer.MAX_VALUE / 4;
private static int N, M;
private static int[][] arr;
public static void main(String[] args) throws IOException {
br = new BufferedReader(new InputStreamReader(System.in));
BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
input();
bw.write(solve());
bw.close();
} // End of main()
private static String solve() {
StringBuilder sb = new StringBuilder();
floyd();
int ans = INF;
for (int i = 1; i <= N; i++) {
for (int j = 1; j <= N; j++) {
if (i == j) continue;
if (arr[i][j] != INF && arr[j][i] != INF) {
ans = Math.min(ans, arr[i][j] + arr[j][i]);
}
}
}
if (ans == INF) {
sb.append(-1);
} else {
sb.append(ans);
}
return sb.toString();
} // End of solve()
private static void floyd() {
for (int i = 1; i <= N; i++) {
for (int j = 1; j <= N; j++) {
for (int k = 1; k <= N; k++) {
if (arr[i][j] > arr[i][k] + arr[k][j]) {
arr[i][j] = arr[i][k] + arr[k][j];
}
}
}
}
} // End of floyd()
private static void input() throws IOException {
StringTokenizer st = new StringTokenizer(br.readLine());
N = Integer.parseInt(st.nextToken());
M = Integer.parseInt(st.nextToken());
arr = new int[N + 1][N + 1];
for (int i = 1; i <= N; i++) {
Arrays.fill(arr[i], INF);
arr[i][i] = 0;
}
for (int i = 0; i < M; i++) {
st = new StringTokenizer(br.readLine());
int a = Integer.parseInt(st.nextToken());
int b = Integer.parseInt(st.nextToken());
int c = Integer.parseInt(st.nextToken());
arr[a][b] = c;
}
} // End of input()
} // End of Main class