洛谷P3371 【模板】单源最短路径（弱化版）-＞P4779 【模板】单源最短路径（标准版）题解

1.原题和原题链接

原题链接弱化版
原题链接标准版

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 的文字位置调换

2.主要思路1

这题就是求单源最短路，由于是弱化版本，数据范围比较小，所以我们可以使用最朴素简单的Dijkstra算法。核心思路就是每次暴力遍历所有点，找出一个离起点最近且没被标记过的点，把它标记为“已确定”，然后用它去更新周围的点的距离。因为要循环N次，每次还要扫一遍N个点找最小值，所以总时间复杂度是O(N^2+m)。虽然慢点，但对于这种小规模数据已经足够解决了，写起来也简单方便。

3.AC代码1

#include<bits/stdc++.h>
using namespace std;
int n,m,s;
int dis[10005],vis[10005];
vector<pair<int,int>> g[10005];
void dj(){
	memset(dis,0x3f,sizeof(dis));
	dis[s]=0;
	for (int i=1;i<=n;i++){
		int mx=0x3f3f3f3f,k=0;
		for (int j=1;j<=n;j++){
			if (vis[j]==0 and dis[j]<mx) mx=dis[j],k=j;
		}vis[k]=1;
		for (int j=0;j<g[k].size();j++){
			int v=g[k][j].first;
			int w=g[k][j].second;
			dis[v]=min(dis[v],dis[k]+w);
		}
	}
}
int main(){
	cin>>n>>m>>s;
	for (int i=0;i<m;i++){
		int x,y,z;cin>>x>>y>>z;
		g[x].push_back({y,z});
	}
	dj();
	for (int i=1;i<=n;i++){
		if (dis[i]==0x3f3f3f3f) cout<<2147483647<<" ";
		else cout<<dis[i]<<" ";
	}
	return 0;
}

接下来我们就要继续完成标准版了。

P4779 【模板】单源最短路径（标准版）

题目背景

2018 年 7 月 19 日，某位同学在 NOI Day 1 T1 归程 一题里非常熟练地使用了一个广为人知的算法求最短路。

然后呢？

$100\to 60$；

$\text{Ag}\to \text{Cu}$；

最终，他因此没能与理想的大学达成契约。

小 F 衷心祝愿大家不再重蹈覆辙。

题目描述

给定一个 $n$ 个点，$m$ 条有向边的带非负权图，请你计算从 $s$ 出发，到每个点的距离。

数据保证你能从 $s$ 出发到任意点。

输入格式

第一行为三个正整数 $n,m,s$。
第二行起 $m$ 行，每行三个非负整数 $u_i,v_i,w_i$，表示从 $u_i$ 到 $v_i$ 有一条权值为 $w_i$ 的有向边。

输出格式

输出一行 $n$ 个空格分隔的非负整数，表示 $s$ 到每个点的距离。

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

说明/提示

样例解释请参考 数据随机的模板题。

$1\le n\le 10^5$；

$1\le m\le 2\times 10^5$；

$s=1$；

$1\le u_i,v_i\le n$；

$0\le w_i\le 10^9$,

$0\le \sum w_i\le 10^9$。

本题数据可能会持续更新，但不会重测，望周知。

2018.09.04 数据更新 from @zzq

2.主要思路2

这题数据大了很多，N增加到了十万，再用O(N^2+m)的暴力方法肯定会TLE。所以这一版的代码我们使用了堆优化的Dijkstra。这里我们用优先队列（小根堆：需要写一个额外的比较器让电脑知道如何同时比较两个数据，即friend bool operator，而大根堆是最大的在上面，所以我们写的比较器就要让小的在上面）来存点，每次直接弹出当前距离最小的点，不用自己暴力找了。弹出来后，如果这个点之前处理过就直接跳过，否则就用它松弛邻边，把更新后的点丢进堆里。这样找最小值的复杂度就降到了O(log⁡ N)，总复杂度变成O((M+N) log N)，跑得更快，也能通过这道题了。

3.AC代码2

#include<bits/stdc++.h>
using namespace std;
int n,m,s,vis[100005],dis[100005];
vector<pair<int,int>> g[100005];
struct node{
	int v,dis;
	friend bool operator < (node n1,node n2){
		return n1.dis>n2.dis;
	}
};
priority_queue<node> pq;
void dj_huojianfeishengban(){
	memset(dis,0x3f,sizeof(dis));
	dis[s]=0;pq.push({s,0});
	while (!pq.empty()){
		int k=pq.top().v;pq.pop();
		if (vis[k]) continue;
		vis[k]=1;
		for (int i=0;i<g[k].size();i++){
			int v=g[k][i].first;
			int w=g[k][i].second;
			if (dis[v]>dis[k]+w){
				pq.push({v,dis[k]+w});
				dis[v]=dis[k]+w;
			}
		}
	}
}
int main(){
	cin>>n>>m>>s;
	for (int i=0;i<m;i++){
		int x,y,z;cin>>x>>y>>z;
		g[x].push_back({y,z});
	}
	dj_huojianfeishengban();
	for (int i=1;i<=n;i++){
		cout<<dis[i]<<" ";
	}
	return 0;
}

改动与优势

主要的改动就是把“暴力找最小值”换成了“优先队列”。原先的代码每次都要遍历所有点找其中dis最小的，太过于浪费时间；而下面代码使用priority_queue(优先队列)来自动维护最小值，取出来只需要O(1)。优势就是速度更快。另外优化后的代码增加了vis数组判断，可以避免重复处理同一个点，效率更高。

以上就是本篇全部内容，感谢浏览！