题目
思路
这道题目就是一个找负环,咱们看一下题目,要求的是从起点回到起点并且回到起点时的权值和是负的,相当于走了一圈路径和是负的,这不就构成了一个负环吗,咱们再来想一下如果起点不在负环中其实也是成立的,只要起点与负环相连就可以了,所以直接bellman_ford算法判断图中有无负环就行了,把穿越虫洞所能倒流的时间取相反数,作为该单向边的权值,最后,有负环就可以,没有就不可以。
题解
#include <iostream>
#include <cstring>
#include <stdio.h>
#include <algorithm>
#include <vector>
using namespace std;
const int INF=0x3f3f3f3f;
const int MAXN=550;
int dist[MAXN];
struct Edge{
int u,v;
int cost;
Edge(int _u=0,int _v=0,int _cost=0):u(_u),v(_v),cost(_cost){}
};
vector<Edge>E;
//点的编号从 1 开始
bool bellman_ford(int start,int n){
for(int i=1;i<=n;i++)dist[i]=INF;
dist[start]=0;
//最多做 n-1 次
for(int i=1;i<n;i++){
bool flag=false;
for(int j=0;j<E.size();j++){
int u=E[j].u;
int v=E[j].v;
int cost=E[j].cost;
if(dist[v]>dist[u]+cost){
dist[v]=dist[u]+cost;
flag=true;
}
}
if(!flag)return true;//没有负环回路
}
for(int j=0;j<E.size();j++)
if(dist[E[j].v]>dist[E[j].u]+E[j].cost)
return false;//有负环回路
return true;//没有负环回路
}
int main()
{
int t;
cin>>t;
for(int i=0;i<t;i++)
{
E.clear();
int n,m,w;
cin>>n>>m>>w;
for(int j=0;j<m;j++)
{
int u,v,cost;
cin>>u>>v>>cost;
Edge a=Edge(u,v,cost);
E.push_back(a);
a=Edge(v,u,cost);
E.push_back(a);
}
for(int j=0;j<w;j++)
{
int u,v,cost;
cin>>u>>v>>cost;
Edge a=Edge(u,v,-cost);
E.push_back(a);
}
if(!bellman_ford(1,n))
{
cout<<"YES"<<endl;
}
else
{
cout<<"NO"<<endl;
}
}
return 0;
}