SPFA算法(最短路径)

Bellman-ford算法的基础上加上一个队列优化,减少了冗余的松弛操作,是一种高效的最短路算法。

设立一个先进先出的队列用来保存待优化的结点,优化时每次取出队首结点u,并且用u点当前的最短路径估计值对离开u点所指向的结点v进行松弛操作,如果v点的最短路径估计值有所调整,且v点不在当前的队列中,就将v点放入队尾。这样不断从队列中取出结点来进行松弛操作,直至队列空为止。

如果某个点进入队列的次数超过N次则存在负环(SPFA无法处理带负环的图)。

#include<iostream>
#include<vector>
#include<deque>
using namespace std;
struct Edge
{
    int to,length;
};
bool spfa(const int &beg,//出发点
          const vector<vector<Edge> > &adjlist,//邻接表,通过传引用避免拷贝
          vector<int> &dist,//出发点到各点的最短路径长度
          vector<int> &path)//路径上到达该点的前一个点
//出现负权回路返回1!
{
    const int &INF=0x7FFFFFFF,&NODE=adjlist.size();
    dist.assign(NODE,INF);//初始化距离为无穷大
    path.assign(NODE,-1);
    deque<int> que(1,beg);
    vector<bool> flag(NODE,0);//标志数组,判断是否在队列中
    vector<int> cnt(NODE,0);//记录各点入队次数,用于判断负权回路
    dist[beg]=0;//出发点到自身路径长度为0
    ++cnt[beg];//开始计数
    flag[beg]=1;//入队
    while(!que.empty())
    {
        const int now=que.front();
        que.pop_front();
        flag[now]=0;//将该点拿出队列
        for(int i=0; i!=adjlist[now].size(); ++i)//遍历所有与当前点有路径的点
        {
            const int &next=adjlist[now][i].to;
            if(dist[now]<INF&&//若距离已知,且优于当前值
                   dist[next]>dist[now]+adjlist[now][i].length)
            {
                dist[next]=dist[now]+adjlist[now][i].length;
                path[next]=now;//记录路径
                if(!flag[next])//若未在处理队列中
                {
                    if(++cnt[next]==NODE)return 1;//计数后出现负权回路
                    if(que.empty()||
                           dist[next]<dist[que.front()])//优先级高于队首
                        que.push_front(next);//放在队首
                    else que.push_back(next);//否则放在队尾
                    flag[next]=1;//入队
                }
            }
        }
    }
    return 0;
}
int main()
{
    int n_num,e_num,beg;//含义见下
    cout<<"输入点数、边数、出发点:";
    cin>>n_num>>e_num>>beg;
    vector<vector<Edge> > adjlist(n_num,vector<Edge>());//默认初始化邻接表
    for(int i=0,p; i!=e_num; ++i)
    {
        Edge tmp;
        cout<<"输入第"<<i+1<<"条边的起点、终点、长度:";
        cin>>p>>tmp.to>>tmp.length;
        adjlist[p].push_back(tmp);
    }
    vector<int> dist,path;//用于接收最短路径长度及路径各点
    if(spfa(beg,adjlist,dist,path))cout<<"图中存在负权回路\n";
    else for(int i=0; i!=n_num; ++i)
        {
            cout<<beg<<"到"<<i<<"的最短距离为"<<dist[i]<<",反向打印路径:";
            for(int w=i; path[w]>=0; w=path[w])
                cout<<w<<"<-";
            cout<<beg<<'\n';
        }
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值