Cod sursa(job #792175)

Utilizator cernat.catallinFMI Cernat Catalin Stefan cernat.catallin Data 26 septembrie 2012 17:59:02
Problema Floyd-Warshall/Roy-Floyd Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.88 kb
#include <fstream>
#include <iostream>
#include <vector>
using namespace std;

vector< vector<unsigned int> > Cost;
int n;

void Read(){
    ifstream f ("royfloyd.in");

    f >> n;
    Cost.resize(n+1, vector<unsigned int>(n+1));
    for(int i = 1; i <= n; ++i)
        for(int j = 1; j <= n; ++j)
            f >> Cost[i][j];
    f.close();
}

void Roy_floyd(){
    for(int k = 1; k <= n; ++k)
        for(int i = 1; i <= n; ++i)
            for(int j = 1; j <= n; ++j)
                if(Cost[i][k] && Cost[k][j] && (Cost[i][j] > Cost[i][k] + Cost[k][j] || !Cost[i][j]) && i!=j) Cost[i][j] = Cost[i][k] + Cost[k][j];
}

void Write(){
    ofstream g ("royfloyd.out");
    for(int i = 1; i <= n; ++i, g << "\n")
        for(int j = 1; j <= n; ++j)
            g << Cost[i][j] << " ";
    g.close();
}

int main(){
    Read();
    Roy_floyd();
    Write();
}