【模板】Floyd的優化

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int dp[105][105];
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    memset(dp,63,sizeof(dp));
    for(int i=1;i<=m;i++)
    {
        int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
        dp[x][y]=min(dp[x][y],z);
        dp[y][x]=min(dp[y][x],z);
    }
    for(int i=1;i<=n;i++) dp[i][i]=0;
    for(int k = 1; k <= n; k++)
    {
        for(int i = 1; i <= n; i++)
        {
            if(k != i)
            {
                int t = (k<i) ? dp[i][k] : dp[k][i];
                if(t == dp[0][0])  continue;
                for(int j = 1; j <= min(k,i); j++)
                {
                    if(dp[i][j]>t+dp[k][j])
                    {
                        dp[i][j]=t+dp[k][j];
                    }
                }
                for(int j = k+1; j <= i; j++)
                {
                    if(dp[i][j]>t+dp[j][k])
                    {
                        dp[i][j]=t+dp[j][k];
                    }
                }
            }
        }
    }
    int s,t;
    scanf("%d%d",&s,&t);
    if(s<t)
    {
        swap(s,t);
    }
    printf("%d\n",dp[s][t]);
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章