Cod sursa(job #2203969)

Utilizator GeorgianBaditaBadita Marin-Georgian GeorgianBadita Data 13 mai 2018 21:31:49
Problema Floyd-Warshall/Roy-Floyd Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.02 kb
#include <fstream>
#define NMAX 120
#define INF 1e9
using namespace std;

ifstream f("royfloyd.in");
ofstream g("royfloyd.out");

int d[NMAX][NMAX];
int n;

void read_data(int &n, int d[][NMAX]){
    f >> n;
    for(int i = 1; i<=n; i++){
        for(int j = 1; j<=n; j++){
            f >> d[i][j];
        }
    }
    for(int i = 1; i<=n; i++){
        for(int j = 1; j<=n; j++){
            if(!d[i][j] && i != j){
                d[i][j] = INF;
            }
        }
    }
}

void roy_floyd(int n, int d[][NMAX]){
    for(int k = 1; k<=n; k++){
        for(int i = 1; i<=n; i++){
            for(int j = 1; j<=n; j++){
                if(d[i][k] + d[k][j] < d[i][j]){
                    d[i][j] = d[i][k] + d[k][j];
                }
            }
        }
    }
}

void write_matrix(int n, int d[][NMAX]){
    for(int i = 1; i<=n; i++){
        for(int j = 1; j<=n; j++){
           g << d[i][j] << ' ';
        }
        g << '\n';
    }
}

int main(){
    read_data(n, d);
    roy_floyd(n, d);
    write_matrix(n, d);

    return 0;
}