题意:一共有n个点。m条双向的路(从a->b要花费t秒,b->a要花费t秒)【权值为正】,
w条单向的路(从a->b可以让时间倒流t秒)【虫洞,可以让时间倒流,相当于该条边的权值是负的】
问:从一个点出发,然后再回到出发的地点。是否不会消耗时间【即判断有无负权值回路】
思路:Bellman-ford算法+判断负权值回路
#include<iostream>
#include<cstring>
using namespace std;
const int MAXN = 5205;//总边数【单向+双向】
const int MAXM = 505;//顶点数
#define INF 0x3f3f3f3f //无穷大
struct Node
{
int u;
int v;
int w;
};
Node edges[MAXN];//邻接表【直接存储边信息】
int dist[MAXM];//v0点到其他顶点的最短路值
int n, m, w, t;//顶点数,双向边数,单向边数,测试数
void Bellman(int v0)//Bellman-ford算法
{
for (int i = 1; i <= n; i++)//初始化
{
dist[i] = INF;
}
dist[v0] = 0;//初始化
for (int i = 2; i < n; i++)从dist(1)[u]递推到dist(2)[u],...,dist(n-1)[u]
{
//判断第i条边<u,v>的引入,是否会缩短v0到v的最短路径
for (int j = 0; j < (2 * m + w); j++)
{ //2 * m + w为总边数
if (dist[edges[j].u] != INF&&dist[edges[j].u] + edges[j].w < dist[edges[j].v])
{
dist[edges[j].v] = dist[edges[j].u] + edges[j].w;
}
}
}
}
bool Negative_loop()//判断是否存在负权值回路
{ //求出distn-1[ ]之后,再对每一条边<u,v>判断一下:
//加入这条边是否会使得顶点v的最短路径值再缩短
for (int j = 0; j < (2 * m + w); j++)
{
if (dist[edges[j].u] != INF&&dist[edges[j].u] + edges[j].w < dist[edges[j].v])
{
return true;//存在从源点可达的负权值回路
}
}
return false;//不存在从源点可达的负权值回路
}
int main()
{
cin >> t;
while (t--)
{
cin >> n >> m >> w;
int i, a, b, val;//数据录入
for (i = 0; i < (2 * m); i += 2)//双向边。
{
cin >> a >> b >> val;
edges[i].u = a;
edges[i].v = b;
edges[i].w = val;
edges[i + 1].u = b;
edges[i + 1].v = a;
edges[i + 1].w = val;
}
for (; i < (2 * m + w); i++)//单向负权值边
{
cin >> a >> b >> val;
edges[i].u = a;
edges[i].v = b;
edges[i].w = -val;
}
Bellman(1);
if (Negative_loop())//若存在负权值回路
{
cout << "YES" << endl;
}
else
{
cout << "NO" << endl;
}
}
return 0;
}