跳到主要内容

Floyd算法

AcWing 854. Floyd求最短路

题目

https://www.acwing.com/problem/content/description/856/ 给定一个 n 个点 m 条边的有向图,图中可能存在重边和自环,边权可能为负数。 再给定 k 个询问,每个询问包含两个整数 xy,表示查询从点 x 到点 y 的最短距离,如果路径不存在,则输出 impossible。 数据保证图中不存在负权回路。

思路

Floyd算法:动态规划的思想

  • 状态表示:f[k][i][j]表示考虑经过前k个点,从ij的距离的最小值
  • 状态计算:f[k][i][j]=f[k-1][i][k]+f[k-1][k][j]

空间可以优化掉一维:f[i][j]=f[i][k]+f[k][j]

代码

#include<bits/stdc++.h>
using namespace std;

typedef long long LL;

const int N=210,INF=0x3f3f3f3f;
int d[N][N];
int n,m,t;

int main()
{
cin>>n>>m>>t;
memset(d,0x3f,sizeof d);
for(int i=1;i<=n;i++) d[i][i]=0;

while(m--)
{
int a,b,c;
cin>>a>>b>>c;
d[a][b]=min(d[a][b],c);
}
//floyd算法
for(int k=1;k<=n;k++)
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
d[i][j]=min(d[i][j],d[i][k]+d[k][j]);

while(t--)
{
int x,y;
cin>>x>>y;
if(d[x][y]>INF/2) cout<<"impossible"<<endl;
else cout<<d[x][y]<<endl;
}
return 0;

}