给定一个n个顶点,m条边的有向图(其中某些边权可能为负,但保证没有负环)。请你计算从1号点到其他点的最短路(顶点从1到n编号)。
第一行两个整数n, m。
接下来的m行,每行有三个整数u, v, l,表示u到v有一条长度为l的边。
1 2 -1
2 3 -1
3 1 2
-2
对于10%的数据,n = 2,m = 2。
对于30%的数据,n <= 5,m <= 10。
对于100%的数据,1 <= n <= 20000,1 <= m <= 200000,-10000 <= l <= 10000,保证从任意顶点都能到达其他所有顶点。
#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
const int MAXN = 20001;
int floyd[MAXN][MAXN];
int main(){
int m, n;
memset(floyd, 2, sizeof(floyd));
cin >> m >> n;
for (int i = 1; i <= m; i++){
int from, to, value;
cin >> from >> to >> value;
floyd[from][to] = value;
}
for (int j = 1; j <= n; j++)
for (int k = 1; k <= n; k++){
if (floyd[1][k] + floyd[k][j] < floyd[1][j])
floyd[1][j] = floyd[1][k] + floyd[k][j];
}
for (int m = 2; m <= n; m++)
cout << floyd[1][m] << endl;
return 0;
}
上网查了一下发现SPFA算法,利用队列优化了一下。
SPFA(Shortest Path Faster Algorithm)(队列优化)算法是求单源最短路径的一种算法,它还有一个重要的功能是判负环(在差分约束系统中会得以体现),在Bellman-ford算法的基础上加上一个队列优化,减少了冗余的松弛操作,是一种高效的最短路算法。
算法大致思路:
s表示源点
利用dist[x]表示从源点s到x的最短距离
用Q队列来保存需要处理的结点
用inQueue[x]保存点x是否在队列中
初始化:dist[]数组全部赋值为无穷大,比如INT_MAX(一定要足够大, 我一开始就是给小了所以有些数据错了)
dist[s] = 0
开始算法:队列+松弛操作
读取Q队首元素并出队(记得把inQueue[Q.top()]置为false)
对与队首结点相连的所有点v进行松弛操作(如果源点通过队首结点再到结点v的距离比源点直接到v的距离要短,就更新dist[v],并且如果inQueue[v] == false 即V当前不在队列中,则v入队,当队列Q为空时,判断结束)
代码如下:
#include<cstdio>
#include<iostream>
#include<cstring>
#include<queue>
using namespace std;
const int MAXN = 20001;
const int MAXL = 200001;
const int INF = INT_MAX;
int dist[MAXN]; //dist[x]表示从源点到x所需的最短距离,初始为INF
int head[MAXN];
int M; //边的索引
bool inQueue[MAXN];
queue<int> Q; //队列Q用来存放可松弛周围结点的结点
struct Edge{
int value;
int to;
int next;
}edge[MAXL]; //采用链式前向星存储边集
//构建边集合
void add(int from, int to, int value){
edge[M].to = to;
edge[M].next = head[from];
edge[M].value = value;
head[from] = M++;
}
//SPFA算法
void SPFA(int start){
dist[start] = 0; //源点到自己的距离为0
Q.push(start);
inQueue[start] = true;
while (!Q.empty()){
int temp = Q.front(); //取队头元素
Q.pop();
for (int j = head[temp]; j != -1; j = edge[j].next){
int toNode = edge[j].to;
if (dist[toNode] > dist[temp] + edge[j].value){ //本题保证无负环,否则需要利用一个数组判断j是否入队超过n次
dist[toNode] = dist[temp] + edge[j].value;
if (!inQueue[toNode]){
Q.push(toNode);
inQueue[toNode] = true;
}
}
}
inQueue[temp] = false;
}
}
int main(){
memset(head, -1, sizeof(head));
int n, m;
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; i++){ //初始化
dist[i] = INF;
inQueue[i] = false;
}
for (int p = 1; p <= m; p++){
int from, to, value;
scanf("%d%d%d", &from, &to, &value); //用cin速度好像要慢一倍= =
add(from, to, value);
}
SPFA(1);
for (int x = 2; x <= n; x++){
printf("%d\n", dist[x]);
}
return 0;
}