1. 程式人生 > 其它 >SPFA最短路

SPFA最短路

題目描述

給定一個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++;
}

void
SPFA() { 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; }