蓝桥杯, 算法训练 最短路 (SPFA)

时间:2023-02-13 10:13:51
问题描述

给定一个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<iostream>
#include<stdio.h>
#include<vector>
#include<queue>
using namespace std;
#define inf 99999999
typedef struct dd
{
int a,d;
}node;
int dis[20005],n;
vector<node>map[20005];
void set()
{
for(int i=1;i<=n;i++)
{
dis[i]=inf;
}
}
void spfa()
{
queue<int>q;
int b[20005]={0},t;
q.push(1); b[1]=1; dis[1]=0;
while(!q.empty())
{
t=q.front(); q.pop();
b[t]=0;
int len=map[t].size(),a;
for(int i=0;i<len;i++)
{
a=map[t][i].a;
if(dis[a]>map[t][i].d+dis[t])
{
dis[a]=map[t][i].d+dis[t];
if(!b[a])
b[a]=1,q.push(a);
}
}

}
}
int main()
{
int m,a,b,l;
node NOW;
while(scanf("%d%d",&n,&m)==2)
{
set();
while(m--)
{
scanf("%d%d%d",&a,&b,&l);
NOW.a=b; NOW.d=l;
map[a].push_back(NOW);
}
spfa();
for(int i=2;i<=n;i++)
printf("%d\n",dis[i]);
}
return 0;
}