Cod sursa(job #1914821)

Utilizator medicinedoctoralexandru medicinedoctor Data 8 martie 2017 18:40:42
Problema Floyd-Warshall/Roy-Floyd Scor 10
Compilator cpp Status done
Runda Arhiva educationala Marime 0.91 kb
#include <fstream>
#include <vector>

using namespace std;

ifstream cin ("royfloyd.in" );
ofstream cout("royfloyd.out");

vector <vector <int> > a;

void read()
{
    int n;
    cin >> n;
    a.resize(n);

    for (int i = 0; i < a.size(); i++)
    {
        a[i].resize(a.size() );

        for (int j = 0; j < a.size(); j++)
            cin >> a[i][j];
    }
}

void solve()
{
    for (int i = 0; i < a.size(); i++)
        for (int j = 0; j < a.size(); j++)
            for (int k = 0; k < a.size(); k++)
                if ((a[i][j] + a[j][k] < a[i][k] && a[i][j] && a[j][k]) || (a[i][k] == 0 && i != k))
                    a[i][k] = a[i][j] + a[j][k];
}

void write()
{
    for (int i = 0; i < a.size(); i++, cout << '\n')
        for (int j = 0; j < a.size(); j++)
            cout << a[i][j] << ' ';
}

int main()
{
    read();

    solve();

    write();

    return 0;
}