接近套用模板的最小生成树算法: bellaman-ford, 只要判断是否存在负环即可。
#include <cstdio>
typedef struct
{
int u;
int v;
int l;
}
path;
path p[5500];
int d[550];
const int inf = 99999999;
int f, n, m, w;
void bellman()
{
bool flag;
for(int i=1; i<=n-1; i++)
{
flag = 0;
for(int j=1; j<=2*m+w; j++)
{
int x = p[j].u;
int y = p[j].v;
if(d[x]!=inf && d[x]+p[j].l < d[y])
{
d[y] = d[x] + p[j].l;
flag = 1;
}
}
if(!flag)
break;
}
flag = 0;
for(int i=1; i<=2*m+w; i++)
if(d[p[i].u] + p[i].l < d[p[i].v])
{
flag = 1;
break;
}
if(flag)
puts("YES");
else
puts("NO");
return;
}
int main()
{
int s, e, t;
scanf("%d", &f);
while(f--)
{
scanf("%d %d %d", &n, &m, &w);
for(int i=1; i<=n; i++)
d[i] = i==1 ? 0 : inf;
for(int i=1; i<=m; i++)
{
scanf("%d %d %d", &s, &e, &t);
p[2*i-1].u = s;
p[2*i-1].v = e;
p[2*i-1].l = t;
p[2*i].u = e;
p[2*i].v = s;
p[2*i].l = t;
}
for(int i=2*m+1; i<=2*m+w; i++)
{
scanf("%d %d %d", &s, &e, &t);
p[i].u = s;
p[i].v = e;
p[i].l = -t;
}
bellman();
}
return 0;
}