Cod sursa(job #2686336)

Utilizator AndreiDeltaBalanici Andrei Daniel AndreiDelta Data 18 decembrie 2020 22:15:16
Problema Floyd-Warshall/Roy-Floyd Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.9 kb
#include <bits/stdc++.h>
#define pb push_back
using namespace std;
ifstream f("royfloyd.in");
ofstream g("royfloyd.out");
const int dim= 100;
const int inf=1e9;
typedef long long ll;
typedef pair<int,int> pi;
int t,T,n,m;

int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    f>>n;
    vector < vector < vector < int > > > dp(2,vector < vector <int > > (n+1,vector < int >(n+1,inf)));
    for(int i=1;i<=n;i++)
    for(int j=1;j<=n;j++)
    {
        f>>dp[0][i][j];
        if(i!=j && dp[0][i][j]==0) dp[0][i][j]=inf;
    }

    for(int k=1;k<=n;k++)
    for(int i=1;i<=n;i++)
    for(int j=1;j<=n;j++)
    if(i!=j)
    dp[k%2][i][j]=min(dp[k%2][i][j],min(dp[(k-1)%2][i][k]+dp[(k-1)%2][k][j],dp[(k-1)%2][i][j]));

    for(int i=1;i<=n;i++,g<<'\n')
    for(int j=1;j<=n;j++)
    if(dp[n%2][i][j]==inf) g<<0<<' ';
    else g<<dp[n%2][i][j]<<' ';

    return 0;
}