Cod sursa(job #1977610)

Utilizator LittleWhoFeraru Mihail LittleWho Data 5 mai 2017 16:58:08
Problema Floyd-Warshall/Roy-Floyd Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.96 kb
#include <bits/stdc++.h>

using namespace std;

int main() {
    freopen("royfloyd.in", "r", stdin);
    freopen("royfloyd.out", "w", stdout);

    int n;
    cin >> n;
    vector< vector<int> > adj_matrix(n, vector<int>(n));
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < n; j++) {
            cin >> adj_matrix[i][j];
        }
    }

    for (int k = 0; k < n; k++) {
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                if (adj_matrix[i][k] && adj_matrix[k][j]
                        && i != j
                        && (adj_matrix[i][j] > adj_matrix[i][k] +adj_matrix[k][j]
                            || !adj_matrix[i][j])) {
                    adj_matrix[i][j] = adj_matrix[i][k] +adj_matrix[k][j];
                }
            }
        }
    }

    for (int i = 0; i < n; i++) {
        for (int j = 0; j < n; j++) {
            cout << adj_matrix[i][j] << " ";
        }
        cout << "\n";
    }

    return 0;
}