[백준 / BOJ] C++ 11780 플로이드 2

11780번: 플로이드 2

 

문제

https://www.acmicpc.net/problem/11780

11780번: 플로이드 2

 

 

풀이

제목부터 노골적으로 플로이드-워셜 알고리즘을 사용하는 문제임을 보여준다. 그리고 각 정점에서 다른 정점으로 갈 때 최단 경로 중 하나를 출력해줘야 하므로 path 배열을 만들어 최단 경로가 갱신될 때마다 i -> k의 경로, k -> j의 경로를 중복되는 k를 제외하고 합쳐준다. 이를 vector의 insert 함수를 사용해서 합쳐서 만들어줬다. 그리고 path를 탐색하며 vector의 size와 원소들을 출력한다.

 

 

코드

#include <bits/stdc++.h>
using namespace std;
#define INF 1e18
#define ll long long

int main() {
  ios::sync_with_stdio(false);
  cin.tie(NULL);
  int N, M;
  ll dist[101][101];
  vector<int> path[101][101];
  cin >> N >> M;
  for (int i = 1; i <= N; i++)
    for (int j = 1; j <= N; j++)
      if (i == j) {
        dist[i][j] = 0;
      } else {
        dist[i][j] = INF;
        path[i][j] = {i, j};
      }
  while (M--) {
    ll a, b, c;
    cin >> a >> b >> c;
    dist[a][b] = min(dist[a][b], c);
  }
  for (int k = 1; k <= N; k++)
    for (int i = 1; i <= N; i++)
      for (int j = 1; j <= N; j++) {
        if (dist[i][j] > dist[i][k] + dist[k][j]) {
          dist[i][j] = dist[i][k] + dist[k][j];
          vector<int> tmp{path[i][k]};
          tmp.insert(tmp.end(), path[k][j].begin() + 1, path[k][j].end());
          path[i][j] = tmp;
        }
      }
  for (int i = 1; i <= N; i++) {
    for (int j = 1; j <= N; j++)
      cout << (dist[i][j] == INF ? 0 : dist[i][j]) << ' ';
    cout << '\n';
  }
  for (int i = 1; i <= N; i++) {
    for (int j = 1; j <= N; j++) {
      if (path[i][j].size() == 0 || dist[i][j] == INF)
        cout << "0\n";
      else {
        cout << path[i][j].size() << ' ';
        for (int e : path[i][j])
          cout << e << ' ';
        cout << '\n';
      }
    }
  }
  return 0;
}