[백준 11404] 플로이드 C++

문제

백준 11404 플로이드 C++

풀이

전형적인 플로이드-와샬 알고리즘 문제이다.

3중 포문을 통해 모든 정점에서부터 모든 정점까지의 최소 거리를 구한다.

소스 코드

#include <iostream>
#include <algorithm>
using namespace std;
 
const int INF = 987654321;
int n, m;
int dist[101][101];
 
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(NULL);
 
    cin >> n;
    cin >> m;
 
    // 시작 지점, 도착 지점, 비용
    int start, end, cost;
 
    // 초기화
    for (int i = 1; i <= n; i++)
    {
        for (int j = 1; j <= n; j++)
        {
            dist[i][j] = INF;
        }
    }
 
    for (int i = 0; i < m; i++)
    {
        cin >> start >> end >> cost;
        dist[start][end] = min(dist[start][end], cost);
    }
 
    // 거쳐가는 노드
    for (int k = 1; k <= n; k++)
    {
        // 시작 노드
        for (int i = 1; i <= n; i++)
        {
            // 도착 노드
            for (int j = 1; j <= n; j++)
            {
                if (i == j)
                    continue;
 
                dist[i][j] = min(dist[i][j], dist[i][k] + dist[k][j]);
            }
        }
    }
 
    for (int i = 1; i <= n; i++)
    {
        for (int j = 1; j <= n; j++)
        {
            if (dist[i][j] == INF)
            {
                cout << 0 << ' ';
            }
            else
            {
                cout << dist[i][j] << ' ';
            }
        }
        cout << '\n';
    }
 
    return 0;
}