题目来源:AcWing 851. spfa求最短路
一、题目描述
给定一个 n n n 个点 m m m 条边的有向图,图中可能存在重边和自环, 边权可能为负数。
请你求出
1
1
1 号点到
n
n
n 号点的最短距离,如果无法从
1
1
1 号点走到
n
n
n 号点,则输出 impossible
。
数据保证不存在负权回路。
输入格式
第一行包含整数
n
n
n 和
m
m
m。
接下来 m m m 行每行包含三个整数 x , y , z x,y,z x,y,z,表示存在一条从点 x x x 到点 y y y 的有向边,边长为 z z z。
输出格式
输出一个整数,表示
1
1
1 号点到
n
n
n 号点的最短距离。
如果路径不存在,则输出 impossible
。
数据范围
1
≤
n
,
m
≤
1
0
5
,
1≤n,m≤10^5,
1≤n,m≤105,
图中涉及边长绝对值均不超过
10000
10000
10000。
输入样例:
3 3
1 2 5
2 3 -3
1 3 4
输出样例:
2
二、SPFA 最短路算法
spfa算法本质是对Bellman-Ford算法的优化,时间复杂度平均
O
(
m
)
O(m)
O(m),最坏情况下
O
(
n
m
)
O(nm)
O(nm),想了解Bellman-Ford算法可以看这里。
spfa算法的时间复杂度是很低的,但是如果出题人卡的话,还是会达到
O
(
n
m
)
O(nm)
O(nm)。
虽然说正权图用Dijkstra算法,但是其实正权图也可以用spfa过掉。
Bellman-Ford算法比较暴力,它每一次迭代都是遍历所有边来尝试更新,但是每一次迭代的话并不是每一条边都需要更新,即 d i s t [ b ] = m i n ( d i s t [ b ] , d i s t [ a ] + w ) dist[b] = min(dist[b], dist[a] + w) dist[b]=min(dist[b],dist[a]+w) 并不是每一次都能成功更新,会造成冗余操作。
而SPFA算法就是针对消除冗余来做优化。如果说 d i s t [ b ] dist[b] dist[b] 在当前迭代下变小了,那么一定是 d i s t [ a ] dist[a] dist[a] 变小了,因此SPFA利用宽搜来做优化,队列里面维护的只有 d i s t dist dist 变小的结点,这样才能有可能成功更新。
核心思路:我更新过谁,我就拿谁更新别人。
伪代码:
queue ← 源点
// 队列里面维护的都是变小的结点,因为只有变小的结点才有可能成功的更新
while (queue 不空)
(1) t ← q.front(); q.pop();
(2) 更新一下 t 的所有出边 t → b
如果说dist[b]更新成功且b不在队列中,queue ← b
三、代码
#include <iostream>
#include <cstring>
using namespace std;
const int N = 1e5 + 10;
int h[N], w[N], e[N], ne[N], idx;
int q[N];
int dist[N];
bool st[N]; // st数组存的是当前结点是否在队列中
int n, m;
void add(int a, int b, int c)
{
w[idx] = c, e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}
int spfa()
{
memset(dist, 0x3f, sizeof dist);
dist[1] = 0;
// 使用循环队列
int hh = 0, tt = 0;
q[tt++] = 1;
st[1] = true;
while (hh != tt)
{
int t = q[hh++];
if (hh == N) hh = 0;
st[t] = false;
for (int i = h[t]; ~i; i = ne[i])
{
int j = e[i];
if (dist[j] > dist[t] + w[i])
{
dist[j] = dist[t] + w[i];
// 只要变小了并且不在队中就入队
// 因为重复的元素入队没有意义
if (!st[j])
{
q[tt++] = j;
if (tt == N) tt = 0;
st[j] = true;
}
}
}
}
if (dist[n] == 0x3f3f3f3f) return -1;
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 == -1) puts("impossible");
else printf("%d\n", t);
return 0;
}