题目描述:
给定一个n个点m条边的有向图,图中可能存在重边和自环, 边权可能为负数。
请你求出1号点到n号点的最短距离,如果无法从1号点走到n号点,则输出impossible。
数据保证不存在负权回路。
输入格式
第一行包含整数n和m。
接下来m行每行包含三个整数x,y,z,表示点x和点y之间存在一条有向边,边长为z。
输出格式
输出一个整数,表示1号点到n号点的最短距离。
如果路径不存在,则输出”impossible”。
数据范围
1≤n,m≤10^5,
图中涉及边长绝对值均不超过10000。
输入样例:
3 3
1 2 5
2 3 -3
1 3 4
输出样例:
2
分析:
Bellman-Ford算法的时间复杂度为O(mn),处理本题显然会超时,而spfa算法则是采用了队列优化后的Bellman-Ford算法,spfa算法一般情况下的时间复杂度为O(m),最坏情况下的时间复杂度为O(mn)。
要理解spfa算法的思路很简单,只需要理解Bellman-Ford算法的缺点即可。回忆下Bellman-Ford算法,每轮松弛都松弛所有的边,在执行过程中可以发现每轮只有很少的边会被松弛成功,因为假设第i轮仅松弛了a节点的距离,那么在下一轮的松弛中,除了a可以到达的点有被松弛的可能性,其它的点不用判断也能知道不会被松弛。既然事先已经知道有些边不会被松弛,那么为什么还要继续去遍历所有边呢?spfa算法的核心思想就是:每一轮只遍历上一轮被松弛的点连接的边,只松弛也可能被松弛的边。
以上图为例,初始情况所有点的距离设为INF,d[A]=0,将A加入队列,出队,松弛A连接的边,更新了d[B]=1,d[C]=5,然后由于B,C均被松弛了,所以也加入队列;第二轮,B出队,更新d[D]=3,d[E]=2,D,E入队,C出队。第三轮,D出队,更新d[F]=5,F入队;E入队,更新d[C]=4,d[G]=4,C,G入队。第四轮,F出队,C出队,G出队,队列为空,算法结束,d[F]=5。
可以看见更新点距离的顺序和Bellman-Ford算法基本一样,要注意的是,出现被松弛的点当其不在队列内才入队,否则仅需更新距离即可,因为队列存储的是点的编号,更新了距离,出队时自然会按照节点最短的距离去执行松弛操作。
#include <iostream>
#include <algorithm>
#include <queue>
#include <vector>
#include <cstring>
using namespace std;
const int maxn = 100005;
typedef pair<int,int> pii;
int n,m,idx;
int d[maxn],h[maxn],e[maxn],ne[maxn],w[maxn];
bool vis[maxn];
queue<int> q;
void add(int a,int b,int c){
e[idx] = b,w[idx] = c,ne[idx] = h[a],h[a] = idx++;//头插法,ne,h数组里存储的均是边的编号
}
int spfa(){
memset(d,0x3f,sizeof d);
d[1] = 0;
q.push(1);
vis[1] = true;
while(q.size()){
int u = q.front();
q.pop();
vis[u] = false;
for(int i = h[u];i != -1;i = ne[i]){
int j = e[i];
if(d[j] > d[u] + w[i]){//只有被松弛的点才可能更新后续点
d[j] = d[u] + w[i];
if(!vis[j]) q.push(j),vis[j] = true;
}
}
}
if(d[n] > 0x3f3f3f3f / 2) return -1;
return d[n];
}
int main(){
scanf("%d%d",&n,&m);
int a,b,c;
memset(h,-1,sizeof h);
while(m--){
scanf("%d%d%d",&a,&b,&c);
add(a,b,c);
}
if(spfa() == -1) cout<<"impossible"<<endl;
else cout<<d[n]<<endl;
return 0;
}