题意:
John的农场里N块地,M条路连接两块地,W个虫洞;路是双向的,虫洞是一条单向路,会在你离开之前把你传送到目的地,
即当你过去的时候,时间会倒退T(s)。我们希望知道会不会在从某块地出发后又回来,看到了离开之前的john自己。
可以将题目简化成,查看图中有没有负权环(belleman-ford算法或者spfa算法均可,我采用的是前者,即belleman-ford算法)。
#include <cstdio>
#include <cstring>
#define INF 10000
#define max 10000
using namespace std;
struct node
{
int s, e, t;
}e[max];
int n, m, w;//m代表双向路条数 w代表单向路条数
int a, b, weight;
int edge = 0;//边数
int dis[max];//距离
int belleman()
{
int flag = 0;
memset(dis, INF, sizeof(dis));//把dis数组全置成INF (依照算法导论上关于Belleman-ford算法部分的讲解)
for (int i = 0;i <= n - 1;i++)
{
flag = 0;
for (int j = 0;j < edge;j++)
{
if (dis[e[j].e]>dis[e[j].s] + e[j].t)
{
dis[e[j].e] = dis[e[j].s] + e[j].t;
flag = 1;
}
}
if (flag == 0)
break;
}
for (int i = 0;i < edge;i++)
if (dis[e[i].e]>dis[e[i].s] + e[i].t)
return false;//存在负环
return true;
}
int main()
{
int casen;
scanf("%d", &casen);
while (casen--)
{
scanf("%d%d%d", &n, &m, &w);
edge = 0;
for (int i = 0;i < m;i++)
{
scanf("%d%d%d", &a, &b, &weight);
e[edge].s = a;
e[edge].e = b;
e[edge++].t = weight;
e[edge].s = b;
e[edge].e = a;
e[edge++].t = weight;
}
for (int i = 0;i < w;i++)
{
scanf("%d%d%d", &a, &b, &weight);
e[edge].s = a;
e[edge].e = b;
e[edge++].t = -weight;
}
if (!belleman())
printf("YES\n");
else
printf("NO\n");
}
}