spfa 算法模板 可求帶負權邊的最短路
阿新 • • 發佈:2018-02-15
cto nbsp init ems push name for 如果 class
它是隊列優化的Bellman-Ford算法。
優化的原理是:下一次松弛操作時被更新dis的點其實與上一次被更新的點有關!如果上一次被更新的點有一條邊指向某點V,那麽在下一次,點V就是可能被更新dis的點。
和 Bellman-Ford 算法一樣,它可以用來求帶負權邊的最短路,如果存在一個從源點可以到達的權重為負值的環路,則返回false表示無解決方案,因為可以不斷在這個環路中循環使總代價越來越小;如果不存在則返回true。
#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; }
spfa 算法模板 可求帶負權邊的最短路