Cod sursa(job #1454103)

Utilizator zsomkoKoman Zsombor zsomko Data 25 iunie 2015 14:10:51
Problema Floyd-Warshall/Roy-Floyd Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.9 kb
#include <fstream>
#include <vector>
using namespace std;

vector<vector<int> > readit(int &n, ifstream &in)
{
    vector<vector<int> > w;
    for(int i=0; i<n; i++)
    {
        w.push_back(vector<int>(n));
        for(int j=0; j<n; j++) 
            in >> w[i][j];
    }
    return w;
}

void royfloyd(int &n, vector<vector<int> > &w)
{
    for(int k=0; k<n; k++)
        for(int i=0; i<n; i++)
            for(int j=0; j<n; j++)
                if(i != j && w[i][k] && w[k][j] && (w[i][j] > w[i][k] + w[k][j] || !w[i][j]))
                    w[i][j] = w[i][k] + w[k][j];
}

int main()
{
    int n;
    ifstream in("royfloyd.in");
    ofstream out("royfloyd.out");
    in >> n;
    vector<vector<int> > w = readit(n, in);
    royfloyd(n, w);
    for(int i=0; i<n; i++)
    {
        for(int j=0; j<n; j++)
            out << w[i][j] << " ";
        out << endl;
    }
    in.close();
    out.close();
    return 0;
}