C++ 最短路（ spfa 判断负环）

spfa 判断负环

题目描述

给定一个 n 个点 m 条边的有向图，图中可能存在重边和自环， 边权可能为负数。

请你判断图中是否存在负权回路。

输入

第一行包含整数 n 和 m。

接下来 m 行每行包含三个整数 x,y,z。表示存在一条从点 x 到点 y 的有向边，边长为 z。

输出

如果图中存在负权回路，则输出 Yes，否则输出 No。

方法一：

#include<bits/stdc++.h>
#define fi first
#define se second
#define pb push_back
#define PII pair<int,int >
#define int long long 
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
  
using namespace std;
 
const int N = 1e6+10;
 
int n,m,A;
int dis[N];
int vis[N]; 
int cnt[N];
vector<pair<int,int >> e[N];
 
int spfa()
{
    for(int i = 1;i<=n;i++) dis[i] = 1e18;
    queue<int > q;
    q.push(1);
    dis[1] = 0;
    vis[1] = 1;
    while(q.size())
    {
        int now = q.front();
        q.pop();
        vis[now] = 0;
        for(auto t:e[now])
        {
            int spot = t.first,w = t.second;
            if(dis[spot]>dis[now]+w)
            {
                cnt[spot] = cnt[now]+1;
                if(cnt[spot]>=n) return false;
                dis[spot] = dis[now]+w;
                if(vis[spot] == 0)
                {
                    vis[spot] = 1;
                    q.push(spot);
                } 
            }
        }
         
    }
    return true;
}
 
signed main()
{
    IOS;
    cin>>n>>m;
    while(m--)
    {
        int a,b,c;
        cin>>a>>b>>c;
        e[a].push_back({b,c});
    }
    spfa()?cout<<"No":cout<<"Yes";
     
    return 0;
}

方法二：

#include<bits/stdc++.h>
#define int long long
 
using namespace std;
 
const int N = 1e5+10;
 
int n,m,A;
int dis[N];
int vis[N];
int cnt[N];
 
vector<pair<int,int> > e[N];
 
int spfa()
{
    for(int i=1;i<=n;i++) dis[i] = 1e18;
    queue<int > q;
    q.push(n+1);
    dis[n+1] = 0;
    vis[n+1] = 1;
    while(q.size())
    {
        int now = q.front();
        q.pop();vis[now] = 0;
        for(auto t:e[now])
        {
            int spot = t.first,w = t.second;
            if(dis[spot]>dis[now]+w)
            {
                cnt[spot] = cnt[now]+1;
                if(cnt[spot]>=n+1) return false;
                dis[spot] = dis[now]+w;
                if(vis[spot]==0)
                {
                    vis[spot] = 1;
                    q.push(spot);
                }
            }
        }
    }
    return true;
}
 
signed main()
{
    cin>>n>>m;
    while(m--)
    {
        int a,b,c;
        cin>>a>>b>>c;
        e[a].push_back({b,c});
    }
    for(int i=1;i<=n;i++)
    {
        e[n+1].push_back({i,0});
    }
    spfa()?cout<<"No":cout<<"Yes";
}

超级原点（有不连通情况时）：

双向边：

for(int i = 1;i<=n;i++)
	{
		e[n+1].push_back({i,0});
		e[i].push_back({n+1,0});
	} 

单向边：

 for(int i=1;i<=n;i++)
    {
        e[n+1].push_back({i,0});
    }

【模板】负环

题目描述

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

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

输入格式

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

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

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

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

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

输出格式

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

#include<bits/stdc++.h>
#define fi first
#define se second
#define pb push_back
#define PII pair<int,int >
#define int long long 
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
 
using namespace std;

const int N = 1e6+10;

int n,m,A;
int dis[N];
int vis[N]; 
int cnt[N];
vector<pair<int,int >> e[N];

int spfa()
{
	queue<int > q;
	q.push(1);
	dis[1] = 0;
	vis[1] = 1;
	while(q.size())
	{
		int now = q.front();
		q.pop();
		vis[now] = 0;
		for(auto t:e[now])
		{
			int spot = t.first,w = t.second;
			if(dis[spot]>dis[now]+w)
			{
				cnt[spot] = cnt[now]+1;
				if(cnt[spot]>=n) return false;
				dis[spot] = dis[now]+w;
				if(vis[spot] == 0)
				{
					vis[spot] = 1;
					q.push(spot);
				} 
			}
		}
		
	}
	return true;
}

signed main()
{
	IOS;
	int t;
	int a,b,c;
	cin>>t;
	while(t--)
	{
		cin>>n>>m;
		for(int i = 1;i<=n;i++)
		{
			e[i].clear();
			dis[i] = 1e18;
			vis[i] = 0;
            cnt[i] = 0;
		}
		while(m--)
		{
			
			cin>>a>>b>>c;
			if(c>=0) 
			{
				e[a].push_back({b,c});
				e[b].push_back({a,c});
			}
			else e[a].push_back({b,c});
		}
		spfa()?cout<<"NO"<<"\n":cout<<"YES"<<"\n";
	}
	
	
	return 0;
}