Cod sursa(job #2198092)

Utilizator horiahoria1Horia Alexandru Dragomir horiahoria1 Data 23 aprilie 2018 17:01:49
Problema Floyd-Warshall/Roy-Floyd Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.93 kb
#include <fstream>
#include <vector>

#define INF (1 << 29)

int main() {
    std::ifstream in("royfloyd.in");
    std::ofstream out("royfloyd.out");

    int i, j, k;

    int n;
    in >> n;

    std::vector<std::vector<int>> d(n, std::vector<int>(n));

    for (i = 0; i < n; ++i) {
        for (j = 0; j < n; ++j) {
            in >> d[i][j];
            if (i != j && d[i][j] == 0) {
                d[i][j] = INF;
            }
        }
    }

    for (k = 0; k < n; ++k) {
        for (i = 0; i < n; ++i) {
            for (j = 0; j < n; ++j) {
                if (i != j) {
                    if (d[i][j] > d[i][k] + d[k][j]) {
                        d[i][j] = d[i][k] + d[k][j];
                    }
                }
            }
        }
    }

    for (i = 0; i < n; ++i) {
        for (j = 0; j < n; ++j) {
            out << d[i][j] << ' ';
        }
        out << '\n';
    }

    in.close();
    out.close();

    return 0;
}