쵼쥬 2022. 2. 8. 10:51


풀이 방법

플로이드 와샬 문제로 먼저 각 도시마다 최소 비용을 구해주었다.

구해주면서 경로를 알아야 하기 때문에 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);
    }
}