【最短路径问题】洛谷 P3371 单源最短路径（弱化版）

单源最短路径 洛谷P3371

算法介绍

• 单源最短路径利用的是SPFA算法，用于计算一个结点到其他所有结点的最短路径。该算法以源点为起始点，不断更新其他点到已经确定距离结点的距离，选取距离最小的结点加入S集合，直到S集合存放有所有的结点
• 看一看题目 洛谷P3371单源最短路径（弱化版）👇

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

题目背景

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

题目描述

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

输入格式

第一行包含三个整数 $n,m,s$，分别表示点的个数、有向边的个数、出发点的编号。

接下来 $m$ 行每行包含三个整数 $u,v,w$，表示一条 $u\to v$ 的，长度为 $w$ 的边。

输出格式

输出一行 $n$ 个整数，第 $i$ 个表示 $s$ 到第 $i$ 个点的最短路径，若不能到达则输出 $2^{31}-1$。

样例 #1

样例输入 #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

提示

【数据范围】
对于 $20\%$ 的数据：$1\le n\le 5$，$1\le m\le 15$；
对于 $40\%$ 的数据：$1\le n\le 100$，$1\le m\le 10^4$；
对于 $70\%$ 的数据：$1\le n\le 1000$，$1\le m\le 10^5$；
对于 $100\%$ 的数据：$1\le n\le 10^4$，$1\le m\le 5\times 10^5$，$1\le u,v\le n$，$w\ge 0$，$\sum w<2^{31}$，保证数据随机。

Update 2022/07/29：两个点之间可能有多条边，敬请注意。

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

样例说明：

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

代码与讲解

• 此题是一道非常经典的最短路径题，而计算这类题目我们一般利用vector数组和queue队列来进行距离的更新
• 具体操作步骤如下

1. 根据题目要求，输入，并建立有向连通图
2. 写SPFA代码，进行图上的点最短路径的更新
3. 依据题意输出最短路径或2^31-1

代码演示（内含注释）

#include<bits/stdc++.h>
#define PII pair<int,int>
#define fi first
#define se second
#define ll long long//long long定义成ll
#define pb push_back
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0)
using namespace std;
const int N = 1e6;
int arr[N],n,m,s,dis[N];
vector <PII> v[N];//建立双元素数组→<点，边权>

void spfa()//SPFA算法
{
	for(int i = 1;i <= n;i++)	dis[i] = 1e9;//初始化，先将每一个元素设成最大值
	queue <int> q;//建立队列
	q.push(s);//压进去
	dis[s] = 0;//将第一个点距离设为0
	while(q.size())
	{
		int now = q.front();//取出最上层坐标
		q.pop();//弹出来
		for(auto t : v[now])//进入下一个队列
		{
			int spot = t.fi;//取出点
			int w = t.se;//取出边权
			if(dis[spot] > dis[now]+w)//当更新的距离比原来更小时
			{
				dis[spot] = dis[now]+w;//更新为小的那个距离
				q.push(spot);//再压进队列
			}
		}
	}
}

int main()
{
	cin>>n>>m>>s;//读入点个数，边条数，出发点编号
	for(int i = 1;i <= m;i++)//建立每一条边
	{
		int u,y,w;
		cin>>u>>y>>w;//读入每条边的两个端点和边长
		v[u].pb({y,w});//存进数组，表示u→y长度为w的有向边
	}
	spfa();//算长度 
	for(int i = 1;i <= n;i++)//输出每一个点的距离
	{
		if(dis[i] == 1e9)	cout<<(1ll<<31)-1<<" ";//如果长度没变，说明到不了这个点，“1LL”表示长整型的1，<<是位运算，表示将二进制原码左移31位，即×2^31
		else				cout<<dis[i]<<" ";//否则输出距离
	}
}

总结

• 此题用了SPFA最短路径算法，类似于动态规划的贪心算法，找到局部最优解，以此来进行原点到每一个点的最短距离的更新。是一道非常经典稍有基础的题。