蓝桥杯 算法提高训练 最短路径
问题描述
给定一个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,保证从任意顶点都能到达其他所有顶点。
我的代码获得70分,运行超时,会有更新。
代码在此:
#include<stdio.h>
#define MAX 0x3f3f3f3f //无穷大
typedef struct Side{
int u,v;
int cost;
}Side;
Side side[200000]; //记录边 200000 * 4 * 3 / 1000000 = 2.4M
int distant[20001]; //从源点到顶点i的路径长度 20000 * 4 /1000000 = 0.08M 加一是因为点从1开始计数
int path[20001]; //记录边的路径 20000 * 4 /1000000 = 0.08M 加一是因为点从1开始计数
int n,m;
int s = 1; //题目要求源点永远为一
int Bellman_Ford () {
int i,j;
int temp;
int re = 1;
Side p; //把全局数据局部化 暂存全局数据
distant[s] = 0; //源点到源点的距离为0
for(i = 2; i <= n; i++){ //除源点的其他点设为无穷大
distant[i] = MAX;
}
for(i = 1; i <= n-1; i++){
for(j = 0; j < m; j++){ //循环所有的边松弛
p = side[j];
temp = distant[p.u] + p.cost;
if(distant[p.v] > temp){
distant[p.v] = temp;
path[p.v] = p.u;
}
}
}
for(j = 0; j < m; j++){ //判断是否出现负环
if(distant[side[j].v] > distant[side[j].u] + side[j].cost){
re = 0;
break;
}
}
return re;
}
int main () {
int i,j;
scanf("%d %d",&n,&m);
for(i = 0; i < m; i ++){
scanf("%d %d %d", &side[i].u, &side[i].v, &side[i].cost);
}
if(Bellman_Ford() == 1){
for(i = 2; i <= n; i++){
printf("%d\n",distant[i]);
}
} else {
printf("error");
}
return 0;
}