spfa求最短路--存在负权边

tech2022-08-21  138

给定一个n个点m条边的有向图,图中可能存在重边和自环, 边权可能为负数。

请你求出1号点到n号点的最短距离,如果无法从1号点走到n号点,则输出impossible。

数据保证不存在负权回路。

输入格式 第一行包含整数n和m。

接下来m行每行包含三个整数x,y,z,表示存在一条从点x到点y的有向边,边长为z。

输出格式 输出一个整数,表示1号点到n号点的最短距离。

如果路径不存在,则输出”impossible”。

数据范围 1≤n,m≤105, 图中涉及边长绝对值均不超过10000。

输入样例: 3 3 1 2 5 2 3 -3 1 3 4 输出样例: 2 存在负权边,没有负权回路(否则将无最短路) spfa适用 思路 其实和Bellman-Ford思路一致,在此基础上的优化 dist[b]=min(dist[b],w[i]+dist[a]); 想一想 只有在dist[a]在这之前变小,dist[b]才有可能被更新。所以可以使用一个队列将变小的点存入,用之去更新其他点(这时将这个点抛出),重复这个操作,直到队列为空。

#include<bits/stdc++.h> using namespace std; const int N=1e4+10; int h[N],w[N],ne[N],idx,e[N]; int add(int a,int b,int c){ e[idx]=b;w[idx]=c;ne[idx]=h[a];h[a]=idx++; } int n,m,k; int dist[N],st[N];//st记录该点是否在队列中,在就不需要重复推入 //dist 记录到i点最短距离 int spfa(){ memset(dist,0x3f,sizeof dist); queue<int> q;q.push(1);dist[1]=0,st[1]=1;//推入起点 记得初始化 while(q.size()){//关键操作 优化 int t=q.front();q.pop();st[t]=0; for(int i=h[t];i!=-1;i=ne[i]){ int j=e[i]; if(dist[j]>dist[t]+w[i]){//这个点变小了,判断是否可以推入队列 dist[j]=dist[t]+w[i]; if(!st[j])q.push(j),st[j]=1; } } } if(dist[n]>0x3f3f3f3f/2) return -1; return dist[n]; } int main(){ cin>>n>>m; memset(h,-1,sizeof(h)); for(int i=0;i<m;i++){ int a,b,c;cin>>a>>b>>c; add(a,b,c); } int t=spfa(); if(t==-1) cout<<"impossible"; else cout<<t; return 0; }

总结 spfa是bellman-ford算法的优化,时间复杂度一般在O(m),最坏情况O(nm); 不光可以使用队列,只要可以应用那个思想都可以。欢迎大家在评论区指出。

最新回复(0)