SPFA算法是基于拓扑排序的升级版,将边权也考虑在内,SPFA实质是一种基于距离更新+队列的算法。
算法描述:基于拓扑排序稍作改动。从给定起点开始,周围点若距离有更新(变长或变短),就更新距离,若被更新点不在队列里,就把被更新点入列;
代码实现
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<ll,ll> pll;
vector<pll> g[100005];
ll n,m,s;
ll dis[100005];
bool vis[100005]; //记录某点是否在队列中
queue<ll> q;
void spfa()
{
memset(dis,0x3f,sizeof(dis));
dis[s]=0;
q.push(s);
vis[s]=1;
while (!q.empty()) {
ll now = q.front(); q.pop();
vis[now]=0;
for(auto to:g[now]) //to是一个pair<ll,ll>
{
ll w=to.second,v=to.first;
if(dis[now]+w<dis[v])//如果找到了一个更小的情况
{
dis[v]=dis[now]+w;
if(vis[v]==0)//如果被更新点不在队列里
{ //就把被更新点加到队列里
q.push(v);
vis[v]=1;
}
}
}
}
}
int main()
{
scanf("%lld%lld%lld",&n,&m,&s);
ll u,v,w;
for( int i = 1 ; i <= m; i++ )
{
scanf("%lld%lld%lld",&u,&v,&w);
g[u].push_back( {v,w} );
}
spfa();
//该输出什么输出什么,不可达则dis[*]==inf
}
return 0;
}
例题
描述
万圣节的晚上,小Hi和小Ho在吃过晚饭之后,来到了一个巨大的鬼屋!
鬼屋中一共有N个地点,分别编号为1…N,这N个地点之间互相有一些道路连通,两个地点之间可能有多条道路连通,但是并不存在一条两端都是同一个地点的道路。
不过这个鬼屋虽然很大,但是其中的道路并不算多,所以小Hi还是希望能够知道从入口到出口的最短距离是多少?
提示:Super Programming Festival Algorithm。
输入
每个测试点(输入文件)有且仅有一组测试数据。
在一组测试数据中:
第1行为4个整数N、M、S、T,分别表示鬼屋中地点的个数和道路的条数,入口(也是一个地点)的编号,出口(同样也是一个地点)的编号。
接下来的M行,每行描述一条道路:其中的第i行为三个整数u_i, v_i, length_i,表明在编号为u_i的地点和编号为v_i的地点之间有一条长度为length_i的道路。
对于100%的数据,满足N<=105,M<=106, 1 <= length_i <= 10^3, 1 <= S, T <= N, 且S不等于T。
对于100%的数据,满足小Hi和小Ho总是有办法从入口通过地图上标注出来的道路到达出口。
输出
对于每组测试数据,输出一个整数Ans,表示那么小Hi和小Ho为了走出鬼屋至少要走的路程。
Sample Input
5 10 3 5
1 2 997
2 3 505
3 4 118
4 5 54
3 5 480
3 4 796
5 2 794
2 5 146
5 4 604
2 5 63
Sample Output
172
AC代码:
#include<cstdio>
#include<iostream>
#include<cstring>
#include<cmath>
#include<queue>
#include<stack>
#include<string>
#include<utility>
#include<algorithm>
#include<map>
#include<sstream>
#include<set>
#define debug(x) cout << "[" << #x <<": " << (x) <<"]"<< endl
#define CLE(a,b) memset(a,b,sizeof(a))
#define MEC(a,b) memcpy(a,b,sizeof(a))
typedef long long ll;
#define inff 0x7fffffff
#define pii pair<int, int>
using namespace std;
typedef pair<ll, ll> pll;
vector<pll> g[1000005];
ll n, m, s, t;
ll dis[100005];
bool vis[100005]; //记录某点是否在队列中
queue<ll> q;
void spfa()
{
memset(dis, 0x3f, sizeof(dis));
dis[s] = 0;
q.push(s);
vis[s] = 1;
while (!q.empty()) {
ll now = q.front(); q.pop();
vis[now] = 0;
for (auto to : g[now]) //to是一个pair<ll,ll>
{
ll w = to.second, v = to.first;
if (dis[now] + w < dis[v])//如果找到了一个更小的情况
{
dis[v] = dis[now] + w;
if (vis[v] == 0)//如果被更新点不在队列里
{ //就把被更新点加到队列里
q.push(v);
vis[v] = 1;
}
}
}
}
}
int main()
{
scanf("%lld%lld%lld%lld", &n, &m, &s, &t);
ll u, v, w;
for (int i = 1; i <= m; i++)
{
scanf("%lld%lld%lld", &u, &v, &w);
g[u].push_back({ v,w });
g[v].push_back({ u,w });
}
spfa();
cout << dis[t] << endl; //未处理不可达得情况
return 0;
}