刷题记录:牛客NC50385虫洞 Wormholes

传送门:牛客

题目描述:

John在他的农场中闲逛时发现了许多虫洞。虫洞可以看作一条十分奇特的有向边，并可以使你
返回到过去的一个时刻（相对你进入虫洞之前）。John的每个农场有M条小路（无向边）连接
着N（从1到N标号）块地，并有W个虫洞。
现在John想借助这些虫洞来回到过去（在出发时刻之前回到出发点），请你告诉他能办到吗。
John将向你提供F个农场的地图。
输入:
2
3 3 1
1 2 2
1 3 4
2 3 1
3 1 3
3 2 1
1 2 3
2 3 4
3 1 8
输出:
NO
YES

一道判环的题目.对于判环,我们可以使用$Spfa$来解决

对于$Spfa$判环,不清楚的可以去这个博客看看,此处就不在赘述了

主要思路:

1. 对于本题来说,我们如果要从一个点出发并且回到之前的时间,这就需要我们经过几条边之后的总权值为负数并且再一次经过出发点,这就意味着存在一个负环,所以本题就可以转化成一道找负环的题目
2. 但是对于本题来说,题目中并没有给出全部的点都是联通的,所以我们需要建一个超级源点,这样才可以保证找负环的正确性

下面是具体的代码部分:

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define root 1,n,1
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
inline ll read() {
	ll x=0,w=1;char ch=getchar();
	for(;ch>'9'||ch<'0';ch=getchar()) if(ch=='-') w=-1;
	for(;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
	return x*w;
}
#define maxn 10000
const double eps=1e-8;
#define	int_INF 0x3f3f3f3f
#define ll_INF 0x3f3f3f3f3f3f3f3f
struct Node{
	int v,w;
};
vector<Node>edge[maxn];
int dis[maxn],vis[maxn];int cnt[maxn];
int n,m,k;
int Spfa(int S) {
	queue<int>q;
	memset(dis,0x3f,sizeof(dis));
	dis[S]=0;
	q.push(S);
	while(!q.empty()) {
		int u=q.front();q.pop();
		vis[u]=0;
		for(int i=0;i<edge[u].size();i++) {
			int v=edge[u][i].v;
			if(dis[v]>dis[u]+edge[u][i].w) {
				dis[v]=dis[u]+edge[u][i].w;
				cnt[v]=cnt[u]+1;
				if(cnt[v]>n) {
					return true;
				}
				if(!vis[v]) {
					q.push(v);
					vis[v]=1;
				}
			}
		}
	}
	return false;
}
int T;
int main() {
	T=read();
	while(T--) {
		memset(vis,0,sizeof(vis));
		memset(cnt,0,sizeof(cnt));
		for(int i=0;i<=n;i++) edge[i].clear();
		n=read();m=read();k=read();
		for(int i=1;i<=m;i++) {
			int u=read(),v=read(),w=read();
			edge[u].push_back({v,w});
			edge[v].push_back({u,w});
		}
		for(int i=1;i<=k;i++) {
			int u=read(),v=read(),w=read();
			edge[u].push_back({v,-w});
		}
		for(int i=1;i<=n;i++) {
			edge[0].push_back({i,0});
		}
		if(Spfa(0)) printf("YES\n");
		else printf("NO\n");
	}
	return 0;
}