暑假-最短路（Bellman-ford、spfa）-F - Wormholes

题意：一共有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;
}