最短路--SPFA算法

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层按上则上升k_i层，按下则下降k_i层，电梯不能到达1层以下，也不能到达n层以上。
输入样例
5 1 5
3 3 1 2 5
0
输出样例
3
解题思路
很多人一看就用dfs和bfs做这道题,但是用SPFA做也是一种很好的方法,把电梯看作一个有向图,在i层上升k_i层可以看作节点i到节点i+k_i的权值为1，表示从i层到k_i层需要按一次按钮，下降也是一样。代码如下：

#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;
}