算法训练 最短路
时间限制:1.0s 内存限制:256.0MB
问题描述
给定一个n个顶点,m条边的有向图(其中某些边权可能为负,但保证没有负环)。请你计算从1号点到其他点的最短路(顶点从1到n编号)。
输入格式
第一行两个整数n, m。
接下来的m行,每行有三个整数u, v, l,表示u到v有一条长度为l的边。
输出格式
共n-1行,第i行表示1号点到i+1号点的最短路。
样例输入
3 3
1 2 -1
2 3 -1
3 1 2
样例输出
-1
-2
数据规模与约定
对于10%的数据,n = 2,m = 2。
对于30%的数据,n <= 5,m <= 10。
对于100%的数据,1 <= n <= 20000,1 <= m <= 200000,-10000 <= l <= 10000,保证从任意顶点都能到达其他所有顶点。
分析:求带负边权的图的最短路径,用SPFA。设置队列存储节点,每当队首点的相邻节点不在队列中时将其入队,直到队空为止。此时各节点到根节点的路径长度即为最短路径。
(没看到题目是有向图,改了好长时间不知道错哪……)
代码:
#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
#define MAXN 20001
#define INF 0x3fffffff
using namespace std;
struct Node {
int v;
int dis;
};
int n, m;
int inqueue[MAXN] = { 0 };
int d[MAXN] = { 0 };
vector<Node> adj[MAXN];
void SPFA() {
fill(inqueue, inqueue + MAXN, 0);
fill(d, d + MAXN, INF);
queue<int> q;
q.push(1);
inqueue[1] = 1;
d[1] = 0;
while (!q.empty()) {
int u = q.front();
q.pop();
inqueue[u] = 0;
for (int i = 0; i < adj[u].size(); i++) {
int v = adj[u][i].v;
int dis = adj[u][i].dis;
if (d[v] > dis + d[u]) {
d[v] = dis + d[u];
if (!inqueue[v]) {
inqueue[v] = 1;
q.push(v);
}
}
}
}
}
int main() {
cin >> n >> m;
for (int i = 0; i < m; i++) {
int u, v, l;
cin >> u >> v >> l;
adj[u].push_back(Node{ v, l });
}
SPFA();
for (int i = 2; i <= n; i++) {
cout << d[i] << endl;
}
return 0;
}