코딩테스트/그래프|트리

[그래프(플로이드)] 백준 11404 플로이드 C++

유(YOO) 2022. 1. 27. 11:15

 

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

 

11404번: 플로이드

첫째 줄에 도시의 개수 n이 주어지고 둘째 줄에는 버스의 개수 m이 주어진다. 그리고 셋째 줄부터 m+2줄까지 다음과 같은 버스의 정보가 주어진다. 먼저 처음에는 그 버스의 출발 도시의 번호가

www.acmicpc.net

모든 정점에서 최단 경로 찾음 O(N^3) 구현 간단

#include <iostream>
#include <algorithm>
#define INF 999999999
#define MAX 501

using namespace std;
int n, m, g[MAX][MAX];

int main(void) {
    cin >> n >> m;
    
    for (int i = 1; i <= n; i++) for (int j = 1; j <= n; j++) g[i][j] = INF;
    for (int i = 1; i <= n; i++) g[i][i] = 0;
    for (int i = 0; i < m; i++) {
        int s, e, w;
        cin >> s >> e >> w;
        if (g[s][e] > w) g[s][e] = w;  // 단방향, 같은 정점 경로 여러 개
    }

    for (int k = 1; k <= n; k++) {
        for (int i = 1; i <= n; i++) {
            for (int j = 1; j <= n; j++) {
                g[i][j] = min(g[i][j], g[i][k] + g[k][j]);
            }
        }
    }

    for (int i = 1; i <= n; i++) {
        for (int j = 1; j <= n; j++) {
            if (g[i][j] == INF) cout << 0 << " ";
            else cout << g[i][j] << " ";
        }
        cout << "\n";
    }

    return 0;
}