[AcWing]851. spfa求最短路（C++实现）spfa算法模板题

1. 题目

2. 读题（需要重点注意的东西）

思路：spfa算法的核心思路是利用队列，当队首元素出队时，更新其所有的相邻节点的值，然后将更新后的节点（只有被更新了的节点才入队）入队，直到队空。

伪代码如下：

3. 解法

---------------------------------------------------解法---------------------------------------------------

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

using namespace std;

const int N = 100010;

int n, m;
int h[N], w[N], e[N], ne[N], idx;
int dist[N];
bool st[N]; // st[]存的是当前这个点是不是在队列中，防止队列中存重复的点

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

int spfa()
{   
    // 初始一下所有点的距离
    memset(dist, 0x3f, sizeof dist);
    dist[1] = 0;

    queue<int> q;
    q.push(1);
    st[1] = true;

    while (q.size())
    {
        int t = q.front();
        
        q.pop();
        st[t] = false;
        
        // 搜索一下t的所有临边
        for (int i = h[t]; i != -1; i = ne[i])
        {
            int j = e[i];
            // 更新所有相邻的节点值
            if (dist[j] > dist[t] + w[i])
            {
                dist[j] = dist[t] + w[i];
                // 如果该节点不在队列中
                if (!st[j])
                {
                    q.push(j);
                    st[j] = true;
                }
            }
        }
    }

    return dist[n];
}

int main()
{
    scanf("%d%d", &n, &m);

    memset(h, -1, sizeof h);

    while (m -- )
    {
        int a, b, c;
        scanf("%d%d%d", &a, &b, &c);
        add(a, b, c);
    }

    int t = spfa();

    if (t == 0x3f3f3f3f) puts("impossible");
    else printf("%d\n", t);

    return 0;
}

---

需要注意的是，dijkstra算法无法解决带负权的问题，是因为再每次用该节点更新完所有边后，当前节点就定死了（即置 st[i] = true），即使之后再发现负权边，也没办法再更新了。
但是spfa算法的st数组，只是用来判断该节点是否在队列中，避免一个节点重复入队，在出队后，其st又会置位false，这是与dijkstra算法的st数组的不同之处。

4. 可能有帮助的前置习题

5. 所用到的数据结构与算法思想

• spfa算法
• 邻接表的存储
• 队列

6. 总结

spfa算法算法模板题，推荐完全背下来。