蓝桥杯 ALGO-5 最短路

  算法训练 最短路  

时间限制：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;
}