[Java] 백준 11404번 플로이드

플로이드 문제입니다.

모든 도시에 가는 최소비용을 구해야합니다.
모든 경로의 비용을 최댓값으로 넣어주고 시작했습니다.

A->B를 가는 경우 
A->C->B를 가는 경우와 비교해서 더욱 빠른 경로로 갱신해주었습니다.
모든 경우에 대해서 갱신을 해주고 결과값을 출력해주었습니다.

주의할점은 초기값이거나 i와 j가 같은경우 0으로 처리해주었습니다.

플로이드 알고리즘은 O(n^3)의 시간복잡도를 가집니다.

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.StringTokenizer;

public class Main {
  public static void main(String[] args) throws Exception {
    BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
    BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

    int n = Integer.parseInt(br.readLine());
    int bus = Integer.parseInt(br.readLine());
    int[][] map = new int[n][n];

    for (int i = 0; i < n; ++i) {
      for (int j = 0; j < n; ++j) {
        map[i][j] = 987654321;
      }
    }

    for (int i = 0; i < bus; ++i) {
      StringTokenizer st = new StringTokenizer(br.readLine());
      int a = Integer.parseInt(st.nextToken()) - 1;
      int b = Integer.parseInt(st.nextToken()) - 1;
      int c = Integer.parseInt(st.nextToken());
      map[a][b] = Math.min(map[a][b], c);
    }

    for (int i = 0; i < n; ++i) {
      for (int j = 0; j < n; ++j) {
        for (int k = 0; k < n; ++k) {
          map[j][k] = Math.min(map[j][k], map[j][i] + map[i][k]);
        }
      }
    }

    for (int i = 0; i < n; ++i) {
      for (int j = 0; j < n; ++j) {
        if (i == j || map[i][j] == 987654321) {
          map[i][j] = 0;
        }
        bw.write(map[i][j] + " ");
      }
      bw.write("\n");
    }

    bw.close();
  }
}