蓝桥杯 算法训练 最短路 （链式前向星，spfa)

算法训练 最短路  

时间限制：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，保证从任意顶点都能到达其他所有顶点。

用Dijkstra会超时，换链式前向星加spfa就过了。用链式前向星时要注意细节啊。

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <queue>
#include <cstring>
#define MAXN 20010
const int inf = 99999999;
struct edge{
	int from, to;
	int w;
};
edge edges[200010];
int dis[MAXN], vis[MAXN];
int first[MAXN], next[200010];
using namespace std;
int main() {
	//freopen("input.txt", "r", stdin);
	int n, m;
	scanf("%d %d", &n, &m);
	int i, j;
	for(i = 0; i <= m; i++) {		//这里不能少了等号 
		first[i] = next[i] = -1;
	}
	for(i = 0; i < m; i++) {
		scanf("%d %d %d", &edges[i].from, &edges[i].to, &edges[i].w);
		next[i] = first[edges[i].from];
		first[edges[i].from] = i;
	}
	for(i = 1; i <= n; i++) {
		dis[i] = inf;
	}
	dis[1] = 0;
	for(i = 1; i <= n; i++) {
		vis[i] = 0;
	}
	queue<int> Q;
	Q.push(1);
	int v;
	while(!Q.empty()) {
		v = Q.front();
		Q.pop();
		vis[v] = 0;
		for(i = first[v]; i != -1; i = next[i]) {
			int to = edges[i].to;
			if(dis[to] > dis[v] + edges[i].w) {
				dis[to] = dis[v] + edges[i].w;
				if(!vis[to]) {
					vis[to] = 1;
					Q.push(to);
				}
			}
		}
	}
	for(i = 2; i <= n; i++) {
		printf("%d\n", dis[i]);
	}
    return 0;
}