Cod sursa(job #798284)

Utilizator toranagahVlad Badelita toranagah Data 16 octombrie 2012 01:35:27
Problema Floyd-Warshall/Roy-Floyd Scor 20
Compilator cpp Status done
Runda Arhiva educationala Marime 1.64 kb
#include <fstream>
using namespace std;

ifstream fin("royfloyd.in");
ofstream fout("royfloyd.out");

int mat[101][101];
int d[101][101];
int N;

int royfloyd(int x, int y);
// int min(int a, int b) {
//     return a < b ? a : b;
// }


int main(int argc, char const *argv[])
{
    fin >> N;
    for (int i = 0; i < N; ++i) {
        for (int j = 0; j < N; ++j) {
            fin >> mat[i][j]; 
        }
    }

    // for (int k = 0; k < N; ++k) {
    //     for (int i = 0; i < N; ++i) {
    //         for (int j = 0; j < N; ++j) {
    //             if (i != j && i != k && j != k && mat[i][k] > 0 && mat[k][j] > 0) {
    //                 if (mat[i][j] > mat[i][k] + mat[k][j] || mat[i][j] == 0) {
    //                     mat[i][j] = mat[i][k] + mat[k][j];
    //                 } 
    //             }
    //         }
    //     }
    // }

    for (int i = 0; i < N; ++i) {
        for (int j = 0; j < N; ++j) {
            mat[i][j] = royfloyd(i, j);
        }
    }
    for (int i = 0; i < N; ++i) {
        for (int j = 0; j < N; ++j) {
            fout << mat[i][j] << ' ';
        }
        fout << '\n';
    }
    return 0;
}

int royfloyd(int x, int y) {
    int result = mat[x][y];
    for (int k = 0; k < N; ++k) {
        if (x != y && x != k && y != k && mat[x][k] > 0 && mat[k][y] > 0) {
            if (result > mat[x][k] + mat[k][y] || result == 0) {
                result = mat[x][k] + mat[k][y];
            } 
        }
    }
    return result;
}

// int royfloyd(int k, int x, int y) {
//     if (k == 0) {
//         return mat[x][y];
//     } else {
//         return min(royfloyd(k-1, x, y), royfloyd(k-1, k, y) + royfloyd(k-1, x, k)
//     }
// }