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