spfa 算法模板 可求带负权边的最短路 判断负环
它是队列优化的Bellman-Ford算法。
优化的原理是:下一次松弛操作时被更新dis的点其实与上一次被更新的点有关!如果上一次被更新的点有一条边指向某点V,那么在下一次,点V就是可能被更新dis的点。
贴一个非常清晰的SPFA算法讲解:链接 ,很遗憾这个讲解没有说如何判断负环是否存在,我补充一下负环的判断方法。
第一种:
在算法更新完状态数组后,再遍历边集,如果存在边还能被松弛的情况,则说明存在负环。
第二种:
如果一个点在被入队次数大于 n 次,那么说明存在负环。原理是虽然一个点在状态数组会被多次更新,但是它的更新次数不会大于n-1次,因为从一个点到另一个点最多经过 n-1 条边!如果存在负环则会造成无限入队的情况,spfa算法陷入死循环,这时候就可直接退出了。
C++代码如下:
#include<iostream> #include<algorithm> #include<cstring> #include<string> #include<set> #include<queue> using namespace std; #define INF 0x3f3f3f3f #define M(a, b) memset(a, b, sizeof(a)) const int maxn = 1000 + 5; struct Edge { int from, to, dist; }; struct SPFA { int d[maxn], cnt[maxn], p[maxn]; int n, m; bool inq[maxn]; vector<int> G[maxn]; vector<Edge> edges; void init(int n) { this->n = n; for (int i = 1; i <= n; ++i) G[i].clear(); edges.clear(); } void AddEdge(int from, int to, int dist) { edges.push_back(Edge{from, to, dist}); int m = edges.size(); G[from].push_back(m-1); } bool spfa(int s) { M(d, INF); M(cnt, 0); M(inq, 0); d[s] = 0; queue<int> q; q.push(s); inq[s] = true; while (!q.empty()) { int u = q.front(); q.pop(); inq[u] = false; for (int i = 0; i < G[u].size(); ++i) { Edge &e = edges[G[u][i]]; if (d[e.to] > d[u] + e.dist) { d[e.to] = d[u] + e.dist; p[e.to] = G[u][i]; if (!inq[e.to]) { q.push(e.to); inq[e.to] = true; if (++cnt[e.to] > n) return false; } } } } return true; } }; SPFA solver; int main() { int n, m, a, b, c; while(cin >> m >> n) { solver.init(n); while(m--) { cin >> a >> b >> c; solver.AddEdge(a, b, c); solver.AddEdge(b, a, c); } solver.spfa(1); cout << solver.d[n] << endl; } return 0; }
————全心全意投入,拒绝画地为牢