Cod sursa(job #2961121)

Utilizator rastervcrastervc rastervc Data 5 ianuarie 2023 20:38:51
Problema Floyd-Warshall/Roy-Floyd Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1 kb
#include <iostream>
#include <fstream>
#include <climits>

using namespace std;

ifstream fin("royfloyd.in");
ofstream fout("royfloyd.out");

constexpr int INF = 1001;
int N, i, j, k, adj[105][105];
int dist[105][105];

static inline void royfloyd() {
    for (i = 1; i <= N; ++i)
        for (j = 1; j <= N; ++j) {
            if (i == j) dist[i][j] = 0;
            else if (adj[i][j]) dist[i][j] = adj[i][j];
            else dist[i][j] = INF;
        }

    for (k = 1; k <= N; ++k)
        for (i = 1; i <= N; ++i)
            for (j = 1; j <= N; ++j)
                dist[i][j] = min(dist[i][j], dist[i][k] + dist[k][j]);
}

int main() {
    fin >> N;
    for (i = 1; i <= N; ++i)
        for (j = 1; j <= N; ++j)
            fin >> adj[i][j];

    royfloyd();

    for (i = 1; i <= N; ++i) {
        for (j = 1; j <= N; ++j) {
            if (dist[i][j] == INF) fout << "0 ";
            else fout << dist[i][j] << ' ';
        }
        fout << '\n';
    }
    return 0;
}