Cod sursa(job #1983445)

Utilizator MaligMamaliga cu smantana Malig Data 22 mai 2017 08:16:56
Problema Floyd-Warshall/Roy-Floyd Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.97 kb
#include <iostream>
#include <fstream>
#include <cstring>

using namespace std;
ifstream in("royfloyd.in");
ofstream out("royfloyd.out");

#define ll long long
#define pb push_back
const int NMax = 1e2 + 5;
const int inf = 1e9 + 5;

int N;
int dist[NMax][NMax];

int main() {
    in>>N;
    for (int i=1;i <= N;++i) {
        for (int j=1;j <= N;++j) {
            int val;
            in>>val;

            dist[i][j] = (i != j && !val) ? inf : val;
        }
    }

    for (int k=1;k <= N;++k) {
        for (int i=1;i <= N;++i) {
            for (int j=1;j <= N;++j) {
                if (dist[i][j] > dist[i][k] + dist[k][j]) {
                    dist[i][j] = dist[i][k] + dist[k][j];
                }
            }
        }
    }

    for (int i=1;i <= N;++i) {
        for (int j=1;j <= N;++j) {
            out<<( (dist[i][j] == inf) ? 0 : dist[i][j] )<<' ';
        }
        out<<'\n';
    }

    in.close();out.close();
    return 0;
}