풀이 방법
플로이드 와샬 문제로 먼저 각 도시마다 최소 비용을 구해주었다.
구해주면서 경로를 알아야 하기 때문에 i -> j로 이동할때 가장 먼저 만나는 도시를 저장해주었다.
가장 먼저 만나는 도시를 이용해서 이동경로를 찾아주었다.
내 코드
package com.company;
import java.io.*;
import java.util.*;
public class Main {
static int N, M;
static int[][][] arr;
static ArrayList<Integer> list = new ArrayList<>();
public static void main(String args[]) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
StringTokenizer st = null;
N = Integer.parseInt(br.readLine());
M = Integer.parseInt(br.readLine());
arr = new int[N + 1][N + 1][2];
for (int i = 1; i <= N; i++) {
for (int j = 1; j <= N; j++) {
arr[i][j][0] = 100000;
}
arr[i][i][0] = 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][0] = Math.min(c, arr[a][b][0]);
arr[a][b][1] = b;
}
for (int i = 1; i <= N; i++) {
for (int j = 1; j <= N; j++) {
for (int k = 1; k <= N; k++) {
if (j == k || k == i)
continue;
if (arr[j][k][0] > arr[j][i][0] + arr[i][k][0]) {
arr[j][k][0] = arr[j][i][0] + arr[i][k][0];
arr[j][k][1] = arr[j][i][1];
}
}
}
}
for (int i = 1; i <= N; i++) {
for (int j = 1; j <= N; j++) {
if (arr[i][j][0] == 100000) {
System.out.print(0 + " ");
} else {
System.out.print(arr[i][j][0] + " ");
}
}
System.out.println();
}
for (int i = 1; i <= N; i++) {
for (int j = 1; j <= N; j++) {
if (i == j || arr[i][j][0] == 100000) {
System.out.println(0);
continue;
}
list.clear();
list.add(i);
dfs(i, j);
}
}
}
static void dfs(int x, int y) {
if (arr[x][y][1] == y) {
list.add(y);
System.out.print(list.size() + " ");
for (int i : list) {
System.out.print(i + " ");
}
System.out.println();
return;
}
list.add(arr[x][y][1]);
dfs(arr[x][y][1], y);
}
}
'코딩테스트 > [백준] 코딩테스트 연습' 카테고리의 다른 글
호텔 - 1106번 (0) | 2022.02.10 |
---|---|
문제 추천 시스템 Version 1 - 21939번 (0) | 2022.02.10 |
최소 회의실 개수 - 19598번 (0) | 2022.02.07 |
가장 긴 바이토닉 부분 수열 - 11054번 (0) | 2022.02.07 |
스타트택시 - 19238번 (0) | 2022.02.07 |