P3385 【模板】负环

题目描述

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

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

输入格式

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

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

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

  • 若 w≥0,则表示存在一条从 u 至 v 边权为 w 的边,还存在一条从 v 至 u 边权为 w 的边。
  • 若 w < 0,则只表示存在一条从 u 至 v 边权为 w 的边。

输出格式

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

输入输出样例

输入

复制代码
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

输出

复制代码
NO
YES

怎么判断负环呢

使用spfa跑一遍最短路,然后用一个数组cnt记录每个点的入队次数,入队次数大于了n,就是有负环,跑了一遍没找到,就没有

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;
const int N=1e5+5;
struct node{
	int to,dis;
};
int T;
vector<node>a[N];
int dis[N];
int vis[N];
int cnt[N];
int n,m;
queue<int>q;
bool spfa(){
	for(int i=1;i<=n;i++)dis[i]=2147483647;
	dis[1]=0;
	q.push(1);
	vis[1]=1;
	while(!q.empty()){
		int x=q.front();
		q.pop();
		vis[x]=0;
		for(int i=0;i<a[x].size();i++){
			int v=a[x][i].to;
			int w=a[x][i].dis;
			if(dis[v]>dis[x]+w){
				dis[v]=dis[x]+w;
				if(vis[v]==0){
					vis[v]=1;
					q.push(v);
					cnt[v]++;
				}
				if(cnt[v]>n)return true;
			}
		}
	}
	return false;
}
signed main(){
	scanf("%d",&T);
	while(T--){
		scanf("%d%d",&n,&m);
		memset(vis,0,sizeof(vis));
		memset(cnt,0,sizeof(vis));
		while(!q.empty())q.pop();
		for(int i=1;i<=n;i++)a[i].clear();
		for(int i=1;i<=m;i++){
			int u,v,w;
			scanf("%d%d%d",&u,&v,&w);
			a[u].push_back(node{v,w});
			if(w>=0)a[v].push_back(node{u,w});
		}
		if(spfa())printf("YES\n");
		else printf("NO\n");
	}
}
相关推荐
逻辑留白陈39 分钟前
Adaboost进阶:与主流集成算法对比+工业级案例+未来方向
算法
Learn Beyond Limits1 小时前
Mean Normalization|均值归一化
人工智能·神经网络·算法·机器学习·均值算法·ai·吴恩达
天选之女wow1 小时前
【代码随想录算法训练营——Day28】贪心算法——134.加油站、135.分发糖果、860.柠檬水找零、406.根据身高重建队列
算法·leetcode·贪心算法
Gohldg1 小时前
C++算法·贪心例题讲解
c++·数学·算法·贪心算法
远远远远子1 小时前
类与对象 --1
开发语言·c++·算法
Aaplloo2 小时前
【无标题】
人工智能·算法·机器学习
西望云天2 小时前
The 2024 ICPC Asia Nanjing Regional Contest(2024南京区域赛EJKBG)
数据结构·算法·icpc
10岁的博客2 小时前
容器化安装新玩法
算法
不会算法的小灰2 小时前
HTML简单入门—— 基础标签与路径解析
前端·算法·html
flashlight_hi3 小时前
LeetCode 分类刷题:1901. 寻找峰值 II
python·算法·leetcode