Cod sursa(job #3153436)

Utilizator trifangrobertRobert Trifan trifangrobert Data 29 septembrie 2023 17:26:22
Problema Floyd-Warshall/Roy-Floyd Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.13 kb
#include <fstream>
#include <vector>
#include <algorithm>

using namespace std;

const int INF = (1 << 30);

int main() {
    ifstream fin("royfloyd.in");
    ofstream fout("royfloyd.out");
    int N;
    fin >> N;
    vector <vector <int>> dp(N, vector <int>(N, INF));
    for (int i = 0;i < N;++i) {
        for (int j = 0;j < N;++j) {
            fin >> dp[i][j];
            if (dp[i][j] == 0) {
                dp[i][j] = INF;
            }
        }
    }
    for (int k = 0;k < N;++k) {
        for (int i = 0;i < N;++i) {
            for (int j = 0;j < N;++j) {
                int dp1 = dp[i][k], dp2 = dp[k][j];
                if (i != j && i != k && j != k && dp1 != INF && dp2 != INF) {
                    dp[i][j] = min(dp[i][j], dp1 + dp2);
                }
            }
        }
    }
    for (int i = 0;i < N;++i) {
        for (int j = 0;j < N;++j) {
            if (dp[i][j] == INF) {
                dp[i][j] = 0;
            }
        }
    }
    for (int i = 0;i < N;++i) {
        for (int j = 0;j < N;++j) {
            fout << dp[i][j] << " \n"[j == N - 1];
        }
    }
    fin.close();
    fout.close();
    return 0;
}