备战CSP（1）：复习图论之最短路算法SPFA

接下来，我们将用这道题目来复习最短路算法，dijk和spfa。

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

关于SPFA

因为spfa实用性更强(个人见解)，所以先复习SPFA。

核心思路

松弛操作：假设出发结点为 S 。

先定义 dis[i] 为S 到 i(当前）的最短路径长度。

操作指：dis[v] = min(dis[v], dis[u]+w(u,v));

算法正确性证明

这个不算dp,因为它有后效性。

但是，它每次松弛操作后，会将v(更新的节点)入队。即让它有了新的可能，以此来影响其他的边。

好了。分析就到这了。开始写代码了。

AC代码

#include<bits/stdc++.h>
using namespace std;
const int MMax = 2147483647;
struct edge{
	int v,w;//v 表示 去的节点，w 表示 权值。 
}; 
int s;//表起点 
int dis[10010],inq[10010];//dis[i]表示 从s 出发 到点i 的最短距离。inq[i]判断节点i有无入队。
vector<edge> g[10010];
int n,m;
void spfa(int s){
	for(int i = 1;i <= n;i++){
		dis[i] = MMax; 
	}//初始化赋值
	queue<int> q;//队列
	q.push(s);
	inq[s] = 1;
	dis[s] = 0;
	while(!q.empty()){
		int u = q.front();//当前访问的节点；
		q.pop();//取出；
		inq[u] = 0;
		for(int i = 0;i < g[u].size();i++){
		    int v = g[u][i].v,w = g[u][i].w;
		    if(dis[u]+w < dis[v]){
		    	dis[v] = dis[u]+w;
		    	if(!inq[v]){
		    		q.push(v);
				    inq[v] = 1;
				}
			}
		} 
	} 
}
int main() {
	cin>>n>>m>>s;
	for(int i = 1;i <= m;i++){
		int u,v,w;
		cin>>u>>v>>w;
		g[u].push_back(edge{v,w});
	}
	spfa(s);
	for(int i =1 ;i <= n;i++){
		cout<<dis[i]<<" ";
	}
}

一遍过，只能说底蕴深厚[doge]。毕竟代码复杂度比线段树、网络流等神犇算法低多了。

关于判断负环

SPFA 也可以通过记录每个点的入队次数判断负环，若有节点入队次数
≥ n，则有负环。这种方式效率会略低。可以参考其他题解。(如果 保证无重边可以记入队次数)

题目

【模板】负环

题目描述

给定一个 $n$ 个点的有向图，请求出图中是否存在从顶点 $1$ 出发能到达的负环。

负环的定义是：一条边权之和为负数的回路。

输入格式

本题单测试点有多组测试数据。

输入的第一行是一个整数 $T$，表示测试数据的组数。对于每组数据的格式如下：

第一行有两个整数，分别表示图的点数 $n$ 和接下来给出边信息的条数 $m$。

接下来 $m$ 行，每行三个整数 $u,v,w$。

• 若 $w\ge 0$，则表示存在一条从 $u$ 至 $v$ 边权为 $w$ 的边，还存在一条从 $v$ 至 $u$ 边权为 $w$ 的边。
• 若 $w<0$，则只表示存在一条从 $u$ 至 $v$ 边权为 $w$ 的边。

输出格式

对于每组数据，输出一行一个字符串，若所求负环存在，则输出 YES，否则输出 NO。

样例 #1

样例输入 #1

2
3 4
1 2 2
1 3 4
2 3 1
3 1 -3
3 3
1 2 3
2 3 4
3 1 -8

样例输出 #1

NO
YES

提示

数据规模与约定

对于全部的测试点，保证：

• $1\le n\le 2\times 10^3$，$1\le m\le 3\times 10^3$。
• $1\le u,v\le n$，$-10^4\le w\le 10^4$。
• $1\le T\le 10$。

提示

请注意，$m$ 不是图的边数。

AC代码

#include<bits/stdc++.h>
using namespace std;
struct edge{
    long long v,w;
}; 
long long n,m,s;
vector<edge> g[110001];
long long dis[1001001],cnt[1001001],inq[1001001];
void spfa(int s){
	for(int i = 1;i <= n;i++){
		dis[i] = 2147483647;
		cnt[i] = 0;
		inq[i] = 0;
	}
	queue<int> que;
	que.push(s);
	inq[s] = 1;
	dis[s] = 0;
	while(!que.empty()){
		int u = que.front();
		que.pop();
		inq[u] = 0; 
		for(int i = 0;i < g[u].size();i++){
			int v = g[u][i].v, w = g[u][i].w;
			if(dis[u]+w< dis[v]){
				dis[v] = dis[u]+w;
				
				if(cnt[v]> n){
					cout<<"YES"<<endl;
					return ;
				}
				if(!inq[v]){
				    cnt[v]++;
					inq[v] = 1;
					que.push(v);
				}
			}
		}
	}
	cout<<"NO"<<endl;
	return ;
}
int main(){
	int t;
	cin>>t;
	while(t--){
	cin>>n>>m;
	for(int i = 1;i <= m;i++){
		int u,v,w;
		cin>>u>>v>>w;
		g[u].push_back(edge{v,w});
		if(w >= 0)g[v].push_back(edge{u,w});
	}
	spfa(1);
	for(int i = 1;i <= n;i++){
		g[i].clear();
	}
    } 
}