SPFA最短路
阿新 • • 發佈:2021-10-21
【題目描述】
給定一個n個點m條邊的有向圖,圖中可能存在重邊和自環,邊權可能為負數。
請你求出1號點到n號點的最短距離,如果無法從1號點走到n號點,則輸出impossible
。
資料保證不存在負權迴路。
輸入格式
第一行包含整數n和m。
接下來mm行每行包含三個整數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的優化版本,Bellman-Ford演算法每次都要鬆弛所有的邊,但很多時候我們其實並不需要那麼多的鬆弛操作,如果一個點被更新了,他才有可能去更新其他的點,因此我們可以使用佇列儲存所有被更新過的點,再拿這些點去更新其他的點,這就是SPFA演算法。
#include <iostream> #include <queue> #include <string.h> using namespace std; const int N = 100009; int h[N],e[N],ne[N],v[N],idx; int n,m; int dist[N],st[N]; queue<int> q; void add(int a,int b,int c) { e[idx] = b; v[idx] = c; ne[idx] = h[a]; h[a] = idx++; } voidSPFA() { memset(dist,0x3f,sizeof dist); dist[1] = 0; q.push(1); st[1] = 1; while(q.size()) { int u = q.front(); q.pop(); st[u] = 0; for(int i = h[u];i != -1;i = ne[i]) { int j = e[i]; if(dist[j] > dist[u] + v[i]) { dist[j]= dist[u] + v[i]; if(!st[j]) { q.push(j); st[j] = 1; } } } } if(dist[n] == 0x3f3f3f3f) cout << "impossible" << endl; else cout << dist[n] << endl; } int main() { memset(h,-1,sizeof h); cin >> n >> m; while(m--) { int a,b,c; cin >> a >> b >> c; add(a,b,c); } SPFA(); return 0; }