蓝桥杯 算法训练 最短路 （链式前向星，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;
}

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
const int inf = 0x3f3f3f3f;
struct node{
	int next, to, w;
};
node edge[200100];
int cnt;
int head[20010];
void add(int u, int v, int w) {
	edge[cnt].to = v;
	edge[cnt].next = head[u];
	edge[cnt].w = w;
	head[u] = cnt++;
}
int dis[20010], vis[20010];
int main() {
	cnt = 0;
	memset(head, -1, sizeof(head));
	int n, m;
	scanf("%d %d", &n, &m);
	int i, j;
	int u, v, w;
	while(m--) {
		scanf("%d %d %d", &u, &v, &w);
		add(u, v, w);
	}
	for(i = 1; i <= n; i++) {
		dis[i] = inf;
		vis[i] = 0;
	}
	queue<int> q;
	q.push(1);
	dis[1] = 0;
	vis[1] = 1;
	while(!q.empty()) {
		int tmp = q.front();
		q.pop();
		for(i = head[tmp]; i != -1; i = edge[i].next) {
			if(dis[edge[i].to] > dis[tmp] + edge[i].w) {
				dis[edge[i].to] = dis[tmp] + edge[i].w;
				if(vis[edge[i].to] == 0) {
					vis[edge[i].to] = 1;
					q.push(edge[i].to);
				}
			}
		}
		vis[tmp] = 0;
	}
	for(i = 2; i <= n; i++) {
		printf("%d\n", dis[i]);
	}
	return 0;
}