题目描述
查找是否有环可以用之前学的那个Dijkstra算法,但y总说如果题目要查询重环,他都是用的spfa.
核心只有一个:cnt靠边数来去操作的,如果有n个点,并且没有环,那么就是n-1条边
如果大于n-1,就是>=n了,那么根据抽屉原理,说明有重点,也就是环!
C++ 代码
#include <cstring>
#include <iostream>
#include <algorithm>
#include <queue>
using namespace std;
const int N = 2010, M = 10010;
int n, m;
int h[N], w[M], e[M], ne[M], idx;
int dist[N], cnt[N];//cnt这个数组存在的原因就是为了去记录点的个数,利用抽屉原理去比较和n的大小
bool st[N];
void add(int a, int b, int c)
{
e[idx] = b, w[idx] = c, ne[idx] = h[a], h[a] = idx ++ ;
}
bool spfa()
{
queue<int> q;
//dist[1] = 0;
//true是被访问过了
for(int i = 1 ; i <= n ; i++)//将点存进去-N;
{
q.push(i);
st[i] = true;
}
while(q.size())
{
int t = q.front();
q.pop();st[t] = false;
for(int i = h[t] ; i != -1 ; i = ne[i])
{
int em = e[i];
if(dist[em] > dist[t] + w[i])
{
dist[em] = dist[t] + w[i];
cnt[em] = cnt[t] + 1;
if(cnt[em] >= n) return true;//cnt靠边数来去操作的,如果有n个点,并且没有环,那么就是n-1条边
//如果大于n-1,就是>=n了,那么根据抽屉原理,说明有重点,也就是环!
if(!st[em])
{
q.push(em);
st[em] = true;
}
}
}
}
return false;
}
int main()
{
scanf("%d%d", &n, &m);
memset(h, -1, sizeof h);
while (m -- )
{
int a, b, c;
scanf("%d%d%d", &a, &b, &c);
add(a, b, c);
}
if (spfa()) puts("Yes");
else puts("No");
return 0;
}