Cod sursa(job #1780671)

Utilizator AdrianaMAdriana Moisil AdrianaM Data 16 octombrie 2016 14:53:10
Problema Floyd-Warshall/Roy-Floyd Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.08 kb
#include <fstream>
#include <vector>
#include <bitset>
#define INF 0x3f3f3f3f
using namespace std;

const string fis = "royfloyd";
ifstream is(fis + ".in");
ofstream os(fis + ".out");

using VI = vector<int>;
using VS = vector<short>;
using VVI = vector<VI>;

int n;
VVI arb;

int main()
{
    is >> n;
    arb = VVI(n + 1, VI(n + 1));
    for ( int i = 1; i <= n; ++i )
        for ( int j = 1; j <= n; ++j )
        {
            is >> arb[i][j];
            if ( !arb[i][j] )
                arb[i][j] = INF;
        }
    for ( int k = 1; k <= n; ++k )
        for ( int i = 1; i <= n; ++i )
            if ( i != k )
                for ( int j = 1; j <= n; ++j )
                    if ( i != j && j != k )
                        arb[i][j] = min(arb[i][j], arb[i][k] + arb[k][j]);
    for ( int i = 1; i <= n; ++i )
    {
        for ( int j = 1; j <= n; ++j )
            if ( arb[i][j] == INF )
                os << "0 ";
            else
                os << arb[i][j] << " ";
        os << "\n";
    }
    is.close();
    os.close();
    return 0;
}