spfa算法

spfa算法是对bellman-ford算法的优化,bellman-ford算法遍历每一条边,不断的迭代更新。但是注意到,有许多边的值是不变的,也就是说不需要更新,所以spfa算法基于这一点进行优化,只有值变小的点才有机会去更新其他点。所以我们用一个队列来存储可以更新其他点的点。

1.初始化
while 队列不空
{
    2.取队头;
    3.标记不在队列中
    4.更新其他点
}

1.初始化需要初始化dis数组为正无穷,将第一个点dis置为0,将第一个点加入队列,并标记在队列中。

2.当队列不空时取队头元素,并将其标记置为不在队列中。

3.需要注意的是spfa中的st数组是标记该点是否在队列中,与dijkstra算法中的st数组作用不同,dijkstra算法中的st数组是在出队时标记已经用过,所以只会标记一次。而spfa数组的元素是可以多次入队的。

4.由于spfa算法是bellman-ford算法的优化,所以spfa算法是可以处理带负权边的单源最短路算法,而dijkstra是只能处理正权边的单源最短路算法。

spfa算法

完整代码:

#include <cstring>
#include <iostream>
#include <algorithm>
#include <queue>

using namespace std;

const int N = 1e5+10;

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

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

int spfa()
{
    memset(dis, 0x3f, sizeof dis);//初始化dis正无穷
    dis[1] = 0;//将第一个点置为0

    queue<int> q;
    q.push(1);//将第一个点加入队列
    st[1] = true;//标记已在队列中

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

        st[t] = false;//从队列中取出,不在队列中

        for (int i = h[t]; i != -1; i = ne[i])
        {
            int j = e[i];
            if (dis[j] > dis[t] + w[i])
            {
                dis[j] = dis[t] + w[i];
                if (!st[j])//只有不在队列中才要将其添加到队列中
                {
                    q.push(j);
                    st[j] = true;
                }
            }
        }
    }

    return dis[n];
}

int main()
{
    cin >> n >> m;

    memset(h, -1, sizeof h);

    while (m -- )
    {
        int a, b, c;
        cin >> a >> b >> c;
        add(a, b, c);
    }

    int t = spfa();

    if (t == 0x3f3f3f3f) puts("impossible");
    else cout << t << endl;

    return 0;
}

相关推荐