[백준 11404] 플로이드 (C++) - 50

문제

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

11404번: 플로이드

생각

굉장히 간단하다.

전에 썼던 플로이드 와샬 알고리즘을 조금만 변형해 금방 풀었다.

플로이드-와샬 알고리즘

void floyd(void) {
    //i->j로 가는 길이 없어도
    //k를 거쳐갈 수 있으면 갈 수 있다고 여긴다
    for (int k = 0; k < N; k++) {
        for (int i = 0; i < N; i++) {
            for (int j = 0; j < N; j++) {
                if (map[i][k] == 1 && map[k][j] == 1)
                    map[i][j] = 1;
            }
        }
    }
}

코드

#include <iostream>
#include <vector>
using namespace std;

const int INF = 987654321;
int n, m;
int map[102][102];

void floyd(void) {
    //i->j로 가는 길이 없어도
    //k를 거쳐갈 수 있으면 갈 수 있다고 여긴다
    for (int k = 1; k <= n; k++) {
        for (int i = 1; i <= n; i++) {
            for (int j = 1; j <= n; j++) {
                if (map[i][k] + map[k][j] < map[i][j])
                    map[i][j] = map[i][k] + map[k][j];
            }
        }
    }
}

int main() {
    ios_base::sync_with_stdio(false);
	cin.tie(0);
    cin >> n >> m;
    for(int i = 1; i <= n; i++) {
        for(int j = 1; j <= n; j++) {
            map[i][j] = INF;
        }
    }
    for(int i = 0; i < m; i++) {
        int s, e, c;
        cin >> s >> e >> c;
        if(map[s][e] > c) map[s][e] = c;
    }

    floyd();

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