P3371 【模板】单源最短路径（弱化版） spfa

P3371 【模板】单源最短路径（弱化版）

题目背景

本题测试数据为随机数据，在考试中可能会出现构造数据让SPFA不通过，如有需要请移步 P4779。

题目描述

如题，给出一个有向图，请输出从某一点出发到所有点的最短路径长度。

输入输出格式

输入格式：

第一行包含三个整数N、M、S，分别表示点的个数、有向边的个数、出发点的编号。

接下来M行每行包含三个整数Fi、Gi、Wi，分别表示第i条有向边的出发点、目标点和长度。

输出格式：

一行，包含N个用空格分隔的整数，其中第i个整数表示从点S出发到点i的最短路径长度（若S=i则最短路径长度为0，若从点S无法到达点i，则最短路径长度为2147483647）

输入输出样例

输入样例#1： 复制

4 6 1
1 2 2
2 3 2
2 4 1
1 3 5
3 4 3
1 4 4

输出样例#1： 复制

0 2 4 3

说明

时空限制：1000ms,128M

数据规模：

对于20%的数据：N<=5，M<=15；

对于40%的数据：N<=100，M<=10000；

对于70%的数据：N<=1000，M<=100000；

对于100%的数据：N<=10000，M<=500000。保证数据随机。

对于真正 100% 的数据，请移步 P4779。请注意，该题与本题数据范围略有不同。

样例说明：

图片1到3和1到4的文字位置调换

代码： 

#include<bits/stdc++.h>
using namespace std;
#define INF 2147483647;
const int maxn=10005;
const int maxm=500005;
int n,m,s,cnt,a,b,c;
int dis[maxn],vis[maxn],head[maxm];
struct node
{
	int x,y,sum,next;
}e[maxm];
void add(int x,int y,int sum)
{
	cnt++;
	e[cnt].x=x;
	e[cnt].y=y;
	e[cnt].sum=sum;
	e[cnt].next=head[x];
	head[x]=cnt;
}//初始化 
void spfa()
{
	queue<int>q;
	for(int i=1;i<=n;i++)
	dis[i]=INF;
	dis[s]=0;
	memset(vis,0,sizeof(vis));
	q.push(s);//入队 
	vis[s]=1;//初始化 
	while(!q.empty())
	{
		int u=q.front();
		q.pop();//出队 
		vis[u]=0;//出队标记 
		for(int i=head[u];i;i=e[i].next)//所有u能走的路 
		{
			int v=e[i].y;
			if(dis[v]>dis[u]+e[i].sum)
			{
				dis[v]=dis[u]+e[i].sum;//取最小的 
				if(vis[v]==0)//如果这个点没在队列里就标记入队 
				{
					vis[v]=1;
					q.push(v);
				}
			}
		} 
	}
}
int main()
{
	cin>>n>>m>>s;
	for(int i=1;i<=m;i++)
	{
		cin>>a>>b>>c;
		add(a,b,c);
	}
	spfa();
	for(int i=1;i<=n;i++)
	cout<<dis[i]<<" ";
	
	return 0;
}