SPFA算法是 Bellman-Ford算法的队列优化版本,实现方式有两种,可以用邻接矩阵实现,也可以用邻接表实现
算法原理
从源点出发依次更新与源点相连的点与源点的最短距离,把更新后的点加入队列,依次遍历队列里的节点,更新节点的邻接点与源点的最短距离.核心代码如下:
邻接矩阵实现
int n,a,b,x,dis[N],w[N][N];//dis数组用于保存每一节点到源点的距离,w数组表示两点之间的距离
int Q[N],head,tail;//Q数组用于构建队列,head为队列头部,tail为队列尾部
bool ex[N];//用于记录节点是否处于队列中
void spfa(int a){
memset(ex,false,sizeof(ex));
memset(dis,-1,sizeof(dis));
head=0,tail=1,Q[1]=a,dis[a]=0,ex[a]=true;
while(head!=tail){
int u=Q[++head];//首部元素出队
ex[u]=false;//出队后ex[u]恢复为false
for(int i=1;i<=n;++i){
//更新最短距离
if(w[u][i]<inf&&(dis[i]==-1||dis[i]>w[u][i]+dis[u])){
dis[i]=w[u][i]+dis[u];
//若队列中没有i点,就把i点加入队列
if(!ex[i]){
Q[++tail]=i;
ex[i]=true;
}
}
}
}
return ;
}
邻接表实现
int num=0;//编号计数器
int head[N]={0};//表头数组
struct edge{
int to;//当前边的终点
int W;//当前边的权值
int nextv;//下一条边的边的编号
}e[N<<1];
int n,a,b,x,dis[N];//dis数组用于保存每一节点到源点的距离,w数组表示两点之间的距离
int Q[N],headl,tail;//Q数组用于构建队列,head为队列头部,tail为队列尾部
bool ex[N];//用于记录节点是否处于队列中
//添加边
void addedge(int v1,int v2,int w){
e[++num].to=v2;
e[num].W=w;
e[num].nextv=head[v1];//上一条边的编号
head[v1]=num;
}
void spfa(int a){
memset(ex,false,sizeof(ex));
memset(dis,-1,sizeof(dis));//初始化到达每一层需要按的最少次数
headl=0,tail=1,Q[1]=a,dis[a]=0,ex[a]=true;
while(headl!=tail){
int u=Q[++headl];//首部元素出队
ex[u]=false;//出队后ex[u]恢复为false
for(int i=head[u];i;i=e[i].nextv){
//更新最短距离
if((dis[e[i].to]==-1||dis[e[i].to]>e[i].W+dis[u])){
dis[e[i].to]=e[i].W+dis[u];
//若队列中没有该点,就把该点加入队列
if(!ex[e[i].to]){
Q[++tail]=e[i].to;
ex[e[i].to]=true;
}
}
}
}
return ;
}
例题
题目链接:HDU1547
题意:输入n,a,b,和长度为n的k数组。共有n层电梯,求从a层到b层至少按几次电梯。电梯只能按向上或者向下,在i层按上则上升ki层,按下则下降ki层,电梯不能到达1层以下,也不能到达n层以上。
输入样例
5 1 5
3 3 1 2 5
0
输出样例
3
解题思路
很多人一看就用dfs和bfs做这道题,但是用SPFA做也是一种很好的方法,把电梯看作一个有向图,在i层上升ki层可以看作节点i到节点i+ki的权值为1,表示从i层到ki层需要按一次按钮,下降也是一样。代码如下:
#include<iostream>
#include<cstdio>
#include<string.h>
#include<vector>
#include<queue>
#include<algorithm>
#define inf 2147483647
#define N 250
#define ll long long
using namespace std;
int n,a,b,x,dis[N],w[N][N];//dis数组用于保存每一节点到源点的距离,w数组表示两点之间的距离
int Q[N],head,tail;//Q数组用于构建队列,head为队列头部,tail为队列尾部
bool ex[N];//用于记录节点是否处于队列中
void spfa(int a){
memset(ex,false,sizeof(ex));
memset(dis,-1,sizeof(dis));//初始化到达每一层需要按的最少次数
head=0,tail=1,Q[1]=a,dis[a]=0,ex[a]=true;
while(head!=tail){
int u=Q[++head];//首部元素出队
ex[u]=false;//出队后ex[u]恢复为false
for(int i=1;i<=n;++i){
//更新最短距离
if(w[u][i]<inf&&(dis[i]==-1||dis[i]>w[u][i]+dis[u])){
dis[i]=w[u][i]+dis[u];
//若队列中没有i点,就把i点加入队列
if(!ex[i]){
Q[++tail]=i;
ex[i]=true;
}
}
}
}
return ;
}
//初始化任意两节点间的距离
void init(){
for(int i=1;i<=n;++i){
for(int j=1;j<=n;++j)
w[i][j]=inf;
}
}
int main()
{
//使输入输出加速
ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
while(cin>>n&&n!=0){
cin>>a>>b;
init();
for(int i=1;i<=n;++i){
cin>>x;
if(i+x<=n)//若电梯能到达i+x或i-x楼,则把两节点距离设为1,表示需要按一次电梯
w[i][i+x]=1;
if(i-x>0)
w[i][i-x]=1;
}
spfa(a);
cout<<dis[b]<<endl;
}
return 0;
}