题目链接:https://vjudge.net/problem/POJ-3259
题意:给你一个双向图以及走一条路所耗费的时间,有一些单向虫洞,能够把人送到t秒之前,问能否看见之前的自己。
思路:判断是否有负圈即可。
#include <cstdio>
#include <algorithm>
#include <queue>
#include <vector>
#include <cstring>
using namespace std;
const int maxn=1005;
const int inf=0x3f3f3f3f;
struct Edge
{
int from,to,dist;
Edge(int u,int v,int d):from(u),to(v),dist(d)
{
}
};
struct SPFA
{
int n,m;
vector <Edge> edges;//保存所有边
vector <int> G[maxn];//只保存边的编号
bool done[maxn];//访问标记
int d[maxn];//到每个点的距离
bool inq[maxn];
int cnt[maxn];
void init(int n)//传参为顶点数
{
this->n=n;
for(int i=0; i<=n; i++)
{
G[i].clear();
}
edges.clear();
}
void addedge(int from,int to,int dist)
{
edges.push_back(Edge(from,to,dist));
m=edges.size();
G[from].push_back(m-1);
}
bool Spfa (int s)
{
queue <int> q;
memset(inq,0,sizeof(inq));
memset(cnt,0,sizeof(cnt));
for(int i=0; i<=n; i++)
d[i]=inf;
d[s]=0;
inq[s]=true;
q.push(s);
while(!q.empty())
{
int u=q.front();
q.pop();
inq[u]=false;
for(int i=0; i<G[u].size(); i++)
{
Edge &e=edges[G[u][i]];
if(d[u]<inf&&d[e.to]>d[u]+e.dist)
{
d[e.to]=d[u]+e.dist;
if(!inq[e.to])
{
q.push(e.to);
inq[e.to]=true;
if(++cnt[e.to]>n)
return false;
}
}
}
}
return true;
}
} spfa;
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
int n,m,w;
scanf("%d%d%d",&n,&m,&w);
spfa.init(n);
for(int i=0; i<m; i++)
{
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
spfa.addedge(a,b,c);
spfa.addedge(b,a,c);
}
for(int i=0; i<w; i++)
{
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
spfa.addedge(a,b,-c);
}
bool flag=false;
for(int i=1; i<=n; i++)
{
if(!spfa.Spfa(i))
{
printf("YES\n");
flag=true;
break;
}
}
if(!flag)
printf("NO\n");
}
return 0;
}