spfa算法模板

bellman-ford算法可以用来求带负权边的最短路,其中关键代码
d[b] = min(d[b],backup[a] + w);,如果对于上一个循环,backup[a]未被更新,这一步也就不需要了。

所以spfa是对bellman-ford算法的优化,思路是对于被更新了的点才进入队列。

复杂度通常为 m ,但最坏时的复杂度仍然是 m * n

模板题:

给定一个 n 个点 m 条边的有向图,图中可能存在重边和自环, 边权可能为负数。

请你求出 1 号点到 n 号点的最短距离,如果无法从 1 号点走到 n 号点,则输出 impossible。

数据保证不存在负权回路。

输入格式
第一行包含整数 n 和 m。

接下来 m 行每行包含三个整数 x,y,z,表示存在一条从点 x 到点 y 的有向边,边长为 z。

输出格式
输出一个整数,表示 1 号点到 n 号点的最短距离。

如果路径不存在,则输出 impossible。

数据范围
1≤n,m≤105,
图中涉及边长绝对值均不超过 10000。

输入样例:
3 3
1 2 5
2 3 -3
1 3 4
输出样例:
2

代码:

#include<bits/stdc++.h>
using namespace std;

const int N = 100010;

int n,m,d[N],st[N];
int e[N],ne[N],w[N],h[N],idx;

void add(int a,int b,int c)
{
    e[idx] = b;
    ne[idx] = h[a];
    w[idx] = c;
    h[a] = idx ++;
}

int spfa()
{
    queue<int>q;
    memset(d,0x3f,sizeof(d));
    d[1] = 0;
    q.push(1);
    st[1] = 1;

    while(q.size())
    {
        int t = q.front();
        q.pop();

        st[t] = 0;

        for(int i = h[t]; i != -1; i = ne[i])
        {
            int j = e[i];
            if(d[j] > d[t] + w[i])
            {
                d[j] = d[t] + w[i];
                if(!st[j])
                {
                    q.push(j);
                    st[j] = 1;
                }
            }
        }
    }
    return d[n];
}

int main()
{
    cin>>n>>m;
    memset(h,-1,sizeof(h));

    for(int i = 1; i <= m; i ++ )
    {
        int a,b,c;
        cin>>a>>b>>c;
        add(a,b,c);
    }

    if(spfa() == 0x3f3f3f3f)
        //不需要dist[n] > 0x3f3f3f3f / 2因为队列里都是由起点更新到的点
        cout<<"impossible";
    else
        cout<<spfa();

    return 0;
}

  • 2
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值