题目大意是在John的农田里有N个区域,分别标记为1....N(1<=N<=500),这个农田里还存在W个虫洞 1<=W<= 200,虫洞是个很神奇的东西,他是一条单向的通道,它会让时间倒流。虫洞的端存在N个区域中。农田里一共有M(1<=M<=2500)条路径,现在告诉你M条路径,以及每条路径的耗用时间,再告诉你W个虫洞的两端位置(单向)以及能让时间倒流多少。现在让你告诉John能否选择一些路径当走完这些路径后可以见到自己。
虫洞的两端让时间倒流,就是将该路径的权赋值为负数就行了,然后用Bellman-Ford寻找图中是否存在负圈。
#ifndef HEAD
#include <stdio.h>
#include <vector>
#include <math.h>
#include <string.h>
#include <string>
#include <iostream>
#include <queue>
#include <list>
#include <algorithm>
#include <stack>
#include <map>
using namespace std;
#endif // !HEAD
#ifndef QUADMEMSET
inline void QuadMemSet(void* dst, int iSize, int value)
{
iSize = iSize / 4;
int* newDst = (int*)dst;
#ifdef WIN32
__asm
{
mov edi, dst
mov ecx, iSize
mov eax, value
rep stosd
}
#else
for (int i = 0; i < iSize; i++)
{
newDst[i] = value;
}
#endif
}
#endif
struct Ege
{
int start;
int end;
int cost;
};
vector<Ege> eges;
int d[501];
bool find_negative_loop(int n,int m)
{
memset(d, 0, sizeof(d));
for (int i = 0; i < n; i++)
{
for (int j = 0; j < m;j++)
{
if (d[eges[j].end] > d[eges[j].start] + eges[j].cost)
{
d[eges[j].end] = d[eges[j].start] + eges[j].cost;
if (i == n - 1)
{
return true;
}
}
}
}
return false;
}
int main()
{
#ifdef _DEBUG
freopen("d:\\in.txt", "r", stdin);
#endif
int F, N, M, W;
scanf("%d\n", &F);
for (int i = 0; i < F;i++)
{
eges.clear();
scanf("%d %d %d\n", &N, &M, &W);
int j = 0;
for (j = 0; j < M; j++)
{
Ege ege;
scanf("%d %d %d\n", &ege.start, &ege.end, &ege.cost);
Ege ege1;
ege1.start = ege.end;
ege1.end = ege.start;
ege1.cost = ege.cost;
eges.push_back(ege);
eges.push_back(ege1);
}
for (; j < W + M;j++)
{
Ege ege;
scanf("%d %d %d\n", &ege.start, &ege.end, &ege.cost);
ege.cost *= -1;
eges.push_back(ege);
}
if (find_negative_loop(N, eges.size()))
{
printf("YES\n");
}
else
{
printf("NO\n");
}
}
return 0;
}