我们只需考虑有向图上的算法,因为无向图是特殊的有向图。我们可以将所有无向边 $u \leftrightarrow v$,都拆分成两条有向边:$u \leftarrow v$ 和 $u \rightarrow v$。
为了方便叙述,我们做如下约定:$n$ 表示图中点数,$m$ 表示图中边数。
图的存储
图一般有两种存储方式:
- 邻接矩阵。开个二维数组,
g[i][j]
表示点 $i$ 和点 $j$ 之间的边权。 - 邻接表。邻接表有两种常用写法,我推荐第二种,代码更简洁,效率也更高,后面有代码模板:
(1) 二维vector:vector<vector<int>> edge
,edge[i][j]
表示第 $i$ 个点的第 $j$ 条邻边。
(2) 数组模拟邻接表:为每个点开个单链表,分别存储该点的所有邻边。
最短路算法
最短路算法分为两大类:
- 单源最短路,常用算法有:
(1) dijkstra,只有所有边的权值为正时才可以使用。在稠密图上的时间复杂度是 $O(n^2)$,稀疏图上的时间复杂度是 $O(mlogn)$。
(2) spfa,不论边权是正的还是负的,都可以做。算法平均时间复杂度是 $O(km)$,$k$ 是常数。 强烈推荐该算法。 - 多源最短路,一般用floyd算法。代码很短,三重循环,时间复杂度是 $O(n^3)$。
算法模板
我们以 poj2387 Til the Cows Come Home 题目为例,给出上述所有算法的模板。
题目大意
给一张无向图,$n$ 个点 $m$ 条边,求从1号点到 $n$ 号点的最短路径。
输入中可能包含重边。
dijkstra算法 $O(n^2)$
最裸的dijkstra算法,不用堆优化。每次暴力循环找距离最近的点。
只能处理边权为正数的问题。
图用邻接矩阵存储。
C++ 代码
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 1010, M = 2000010, INF = 1000000000;
int n, m;
int g[N][N], dist[N]; // g[][]存储图的邻接矩阵, dist[]表示每个点到起点的距离
bool st[N]; // 存储每个点的最短距离是否已确定
void dijkstra()
{
for (int i = 1; i <= n; i++) dist[i] = INF;
dist[1] = 0;
for (int i = 0; i < n; i++)
{
int id, mind = INF;
for (int j = 1; j <= n; j++)
if (!st[j] && dist[j] < mind)
{
mind = dist[j];
id = j;
}
st[id] = 1;
for (int j = 1; j <= n; j++) dist[j] = min(dist[j], dist[id] + g[id][j]);
}
}
int main()
{
cin >> m >> n;
for (int i = 1; i <= n; i++)
for (int j = 1; j <= n; j++)
g[i][j] = INF;
for (int i = 0; i < m; i++)
{
int a, b, c;
cin >> a >> b >> c;
g[a][b] = g[b][a] = min(g[a][b], c);
}
dijkstra();
cout << dist[n] << endl;
return 0;
}
dijkstra+heap优化 $O(mlogn)$
用堆维护所有点到起点的距离。时间复杂度是 $O(mlogn)$。
这里我们可以手写堆,可以支持对堆中元素的修改操作,堆中元素个数不会超过 $n$。也可以直接使用STL中的priority_queue
,但不能支持对堆中元素的修改,不过我们可以将所有修改过的点直接插入堆中,堆中会有重复元素,但堆中元素总数不会大于 $m$。
只能处理边权为正数的问题。
图用邻接表存储。
C++ 代码
typedef pair<int, int> PII;
int n; // 点的数量
int h[N], w[N], e[N], ne[N], idx; // 邻接表存储所有边
int dist[N]; // 存储所有点到1号点的距离
bool st[N]; // 存储每个点的最短距离是否已确定
// 求1号点到n号点的最短距离,如果不存在,则返回-1
int dijkstra()
{
memset(dist, 0x3f, sizeof dist);
dist[1] = 0;
priority_queue<PII, vector<PII>, greater<PII>> heap;
heap.push({0, 1}); // first存储距离,second存储节点编号
while (heap.size())
{
auto t = heap.top();
heap.pop();
int ver = t.second, distance = t.first;
if (st[ver]) continue;
st[ver] = true;
for (int i = h[ver]; i != -1; i = ne[i])
{
int j = e[i];
if (dist[j] > distance + w[i])
{
dist[j] = distance + w[i];
heap.push({dist[j], j});
}
}
}
if (dist[n] == 0x3f3f3f3f) return -1;
return dist[n];
}
spfa算法 $O(km)$
bellman-ford算法的优化版本,可以处理存在负边权的最短路问题。
最坏情况下的时间复杂度是 $O(nm)$,但实践证明spfa算法的运行效率非常高,期望运行时间是 $O(km)$,其中 $k$ 是常数。
但需要注意的是,在网格图中,spfa算法的效率比较低,如果边权为正,则尽量使用 dijkstra 算法。
图采用邻接表存储。
队列为手写的循环队列。
C++ 代码
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <queue>
using namespace std;
const int N = 1010, M = 2000010, INF = 1000000000;
int n, m;
int dist[N], q[N]; // dist表示每个点到起点的距离, q 是队列
int h[N], e[M], v[M], ne[M], idx; // 邻接表
bool st[N]; // 存储每个点是否在队列中
void add(int a, int b, int c)
{
e[idx] = b, v[idx] = c, ne[idx] = h[a], h[a] = idx++;
}
void spfa()
{
int hh = 0, tt = 0;
for (int i = 1; i <= n; i++) dist[i] = INF;
dist[1] = 0;
q[tt++] = 1, st[1] = 1;
while (hh != tt)
{
int t = q[hh++];
st[t] = 0;
if (hh == n) hh = 0;
for (int i = h[t]; i != -1; i = ne[i])
if (dist[e[i]] > dist[t] + v[i])
{
dist[e[i]] = dist[t] + v[i];
if (!st[e[i]])
{
st[e[i]] = 1;
q[tt++] = e[i];
if (tt == n) tt = 0;
}
}
}
}
int main()
{
memset(h, -1, sizeof h);
cin >> m >> n;
for (int i = 0; i < m; i++)
{
int a, b, c;
cin >> a >> b >> c;
add(a, b, c);
add(b, a, c);
}
spfa();
cout << dist[n] << endl;
return 0;
}
floyd算法 $O(n^3)$
标准弗洛伊德算法,三重循环。循环结束之后 $d[i][j]$ 存储的就是点 $i$ 到点 $j$ 的最短距离。
需要注意循环顺序不能变:第一层枚举中间点,第二层和第三层枚举起点和终点。
由于这道题目的数据范围较大,点数最多有1000个,因此floyd算法会超时。
但我们的目的是给出算法模板哦~
C++ 代码
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <queue>
using namespace std;
const int N = 1010, M = 2000010, INF = 1000000000;
int n, m;
int d[N][N]; // 存储两点之间的最短距离
int main()
{
cin >> m >> n;
for (int i = 1; i <= n; i++)
for (int j = 1; j <= n; j++)
d[i][j] = i == j ? 0 : INF;
for (int i = 0; i < m; i++)
{
int a, b, c;
cin >> a >> b >> c;
d[a][b] = d[b][a] = min(c, d[a][b]);
}
// floyd 算法核心
for (int k = 1; k <= n; k++)
for (int i = 1; i <= n; i++)
for (int j = 1; j <= n; j++)
d[i][j] = min(d[i][j], d[i][k] + d[k][j]);
cout << d[1][n] << endl;
return 0;
}
头晕
##### 0x3f3f3f3f与0x3f不一样吗??
请问这里的add函数中那几个数组存的都是什么啊
就是数组模拟单链表,可以参考826. 单链表。
有大佬帮小白解释一下 第一个代码里的倒数第四行 g[a][b] = g[b][a] = min(g[a][b], c); 这句什么意思吗?
应该是输入中存在重复边,取权值最小的当做边权值。如不对,还请指正^_^
对滴。
y总,想问一下,那个邻接矩阵 g[ ] [ ],什么时候全部初始化为 INF,什么时候需要 i==j 时 初始化为 g[i][j] = 0,有点迷糊==.==
具体情况具体分析。
为啥heap优化的dijkstra没有st数组
已更正~
太棒了
谢谢hh
y总,你的模板过模板题都有问题q w q
哪个模板有问题
s p e a
无解情况一般需要根据具体问题自己特判。
关于堆优化版的dijkstra算法时间复杂度,我看到其他博客写的都是O((m+n)logn),而大佬你的是O(mlogn),是不是模板不同?
一般情况下点数 $n$ 均小于等于边数 $m$,所以 $O((m+n)logn)$ 和 $O(mlogn)$ 是一样的。
大佬能不能写一下每次定义的一些变量数组代表得什么呀
已加注释~
大雪菜威武,献上我的膝盖,跟着大佬学算法
好棒!
请问 dijkstra_heap算法中,已经有了
if (dist[e[i]] > t.first + v[i])
使得不会有点重复入堆,为什么还需要if (t.first > dist[t.second]) continue;
?自答一下:dijkstra要求每个点之被遍历到一次。
这个
if (t.first > dist[t.second]) continue;
就是避免同一个点(假设为A)被重复遍历到。假设C指向A,B也指向A。假设C先被遍历,在遍历到C时A也会入堆;然后又遍历到B,通过B到A的路径更短,所以通过C到A的这条数据就不必被考虑了。即使有了
if (dist[e[i]] > t.first + v[i])
这句话,还是有可能会有重复元素入堆的,楼上同学已经给出了一个例子hh堆优化的dijkstra里
if (t.first > dist[t.second]) continue;
是不是应该换成if (t.first != dist[t.second]) continue;
不然对于负权图就没法保证每个点只处理一次了只有当图的边权都是正的时,dijkstra算法才是正确的。如果存在负权,一般用spfa算法。另外,其实
if (t.first != dist[t.second])
这种判断方式,当存在权值为零的边时,仍然不能保证每个点只处理一次,此时可以开一个bool
数组,来标记每个点是否已经被处理过。哦哦,如果用
if (t.first > dist[t.second]) continue;
来判断,这个算法和spfa的区别在哪里呢?貌似他们都可以对负权图产生正确的结果。因为对于一个点,这个点的最优值可能会再次被后来更长的路径给替代(因为有负权),之后会把这个点再次压入堆中,然后再处理一遍这个点的邻居点,也就是每个点会被处理多次直到找到最优值?如果允许每个点更新多次,其实它就不是dijkstra算法了,相当于将spfa算法中的队列换成堆。下面我来说一下为何一般在spfa算法中,不用堆来替换队列:
对于稀疏图而言,正权图用dijkstra算法,是因为它的最坏时间复杂度是 $O(mlogn)$;负权图用spfa算法,一般情况下的时间复杂度是 $O(m)$,最坏是 $O(nm)$,如果把spfa里的队列换成堆(也就是上面提到的堆优化dijkstra处理负权图),那么时间复杂度会变成 一般情况下是 $O(mlogn)$ 和 $O(nmlogn)$。所以队列比堆要好。
明白了,感谢大佬的讲解~👍
弱弱的问一下,可以转载么??
可以,注明出处即可~
请问一下,poj这题,为什么把g[a][b] = g[b][a] = min(c, g[a][b])换成g[a][b]=g[b][a]=c就会WA
数据中可能存在重边,我们应该只保留权值最小的边。
原来这样 谢谢了
客气啦
spfa不是容易被卡掉吗
绝大部分情况不会被卡。如果边权都是非负的,为了保险可以写dijkstra;如果边权有正有负,建议写spfa。
skr
超赞!
谢谢!