最短路【SPFA】

来源:互联网 发布:pytorch支持windows吗 编辑:程序博客网 时间:2024/06/06 18:10
问题描述

给定一个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,保证从任意顶点都能到达其他所有顶点。

#include<stdio.h>#include<string.h>#include<algorithm>#define INF 0x3f3f3f3fusing namespace std;int next[200010] , f[200010], vis[20010], n, m, dis[20010], que[200010];struct point{int u, v, w;}p[200010];int min(int x, int y){return x<y?x:y;}void spfa(){memset(vis, 0, sizeof(vis));memset(dis, 0, sizeof(dis));vis[1] = 1;for(int i = 2; i <= n; ++i)dis[i] = INF;int front = 0, rear = 0;que[rear++] = 1;while(front < rear){int u = que[front++];vis[u] = 0;for(int i = f[u]; i != -1; i = next[i]){int v = p[i].v;if(dis[v] > dis[u]+p[i].w){dis[v] = dis[u]+p[i].w;if(!vis[v]){que[rear++] = v;vis[v] = 1;}}} } }int main(){int i, j, k;while(scanf("%d%d", &n, &m) != EOF){memset(f, -1, sizeof(f));memset(next, -1, sizeof(next));for(i = 0; i < m; ++i){scanf("%d%d%d", &p[i].u, &p[i].v, &p[i].w);next[i] = f[p[i].u];f[p[i].u] = i;}spfa();for(i = 2; i <= n; ++i)printf("%d\n", dis[i]); }return 0;} 

题意:就是给一个图求最短路。

思路:用djs+邻接矩阵超时了,所以用了spfa+邻接表。

0 0
原创粉丝点击