题目描述:
给定一个n个点m条边的有向图,图中可能存在重边和自环, 边权可能为负数。
请你判断图中是否存在负权回路。
输入格式
第一行包含整数n和m。
接下来m行每行包含三个整数x,y,z,表示点x和点y之间存在一条有向边,边长为z。
输出格式
如果图中存在负权回路,则输出“Yes”,否则输出“No”。
数据范围
1≤n≤2000,
1≤m≤10000,
图中涉及边长绝对值均不超过10000。
输入样例:
3 3
1 2 -1
2 3 4
3 1 -4
输出样例:
Yes
分析:
在Bellman-Ford算法中,我们知道,经过最大n-1轮松弛操作,从起点到所有点的最短距离就被确定下了。而图中若存在负权回路,则每次通过该负权回路来松弛距离时,最短距离都会减小,是不存在最短路径的。所以可以在spfa算法中用cnt数组存储当前点最短路径经过的边数,一旦经过了n条边,则说明经过了n + 1个点,而图中仅有n个点,该最短路径经过了一个点两次,所以必然存在负权回路。
#include <iostream>
#include <algorithm>
#include <queue>
#include <vector>
#include <cstring>
using namespace std;
const int maxn = 2005,maxm = 10005;
typedef pair<int,int> pii;
int n,m,idx;
int d[maxn],h[maxn],cnt[maxn],e[maxm],ne[maxm],w[maxm];
bool vis[maxn];
queue<int> q;
void add(int a,int b,int c){
e[idx] = b,w[idx] = c,ne[idx] = h[a],h[a] = idx++;//头插法,ne,h数组里存储的均是边的编号
}
bool spfa(){
for(int i = 1;i <= n;i++){//全放进去
q.push(i);
vis[i] = true;
}
while(q.size()){
int u = q.front();
q.pop();
vis[u] = false;
for(int i = h[u];i != -1;i = ne[i]){
int j = e[i];
if(d[j] > d[u] + w[i]){//只有被松弛的点才可能更新后续点
d[j] = d[u] + w[i];
cnt[j] = cnt[i] + 1;
if(cnt[j] >= n) return true;
if(!vis[j]) q.push(j),vis[j] = true;
}
}
}
return false;
}
int main(){
scanf("%d%d",&n,&m);
int a,b,c;
memset(h,-1,sizeof h);
while(m--){
scanf("%d%d%d",&a,&b,&c);
add(a,b,c);
}
if(spfa()) cout<<"Yes"<<endl;
else cout<<"No"<<endl;
return 0;
}