同上题,不过改的spfa算法,注意每个节点进入队列的次数至多为n-1次(一共n个节点),若进入大于等于n次了,则说明图中存在负权回路,此时正好满足题目中时光倒流的要求,另外注意,vector每次用的时候清空。
下面是spfa算法的简单说明:
我们用数组d记录每个结点的最短路径估计值,而且用邻接表来存储图G。我们采取的方法是松弛:设立一个先进先出的队列用来保存待优化的结点,优化时每次取出队首结点u,并且用u点当前的最短路径估计值对离开u点所指向的结点v进行松弛操作,如果v点的最短路径估计值有所调整,且v点不在当前的队列中,就将v点放入队尾。这样不断从队列中取出结点来进行松弛操作,直至队列空为止。
[color=red]注意对刚出队列的节点的所有相邻节点都要做松弛操作,不管该节点是否在队列中,不在队列中的松弛点加入到队列即可。[/color]
下面是spfa算法的简单说明:
我们用数组d记录每个结点的最短路径估计值,而且用邻接表来存储图G。我们采取的方法是松弛:设立一个先进先出的队列用来保存待优化的结点,优化时每次取出队首结点u,并且用u点当前的最短路径估计值对离开u点所指向的结点v进行松弛操作,如果v点的最短路径估计值有所调整,且v点不在当前的队列中,就将v点放入队尾。这样不断从队列中取出结点来进行松弛操作,直至队列空为止。
[color=red]注意对刚出队列的节点的所有相邻节点都要做松弛操作,不管该节点是否在队列中,不在队列中的松弛点加入到队列即可。[/color]
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
#define INF 999999999
using namespace std;
struct E
{
int v;
int w;
}edges[6000];
vector<E> v[1008];
bool visited[1008];
int dist[1008];
int enterCount[1008];
int n,m,w,edgeCount;
void addEdge(int e,int s,int t)
{
edgeCount++;
edges[edgeCount].v=s;
edges[edgeCount].w=t;
v[e].push_back(edges[edgeCount]);
}
bool spfa()
{
int i;
queue<int> q;
visited[1]=1;
dist[1]=0;
enterCount[1]=1;
for(i=2;i<=n;i++)
{
visited[i]=0;
dist[i]=INF;
enterCount[i]=0;
}
q.push(1);
while(!q.empty())
{
int temp=q.front();
q.pop();
visited[temp]=0;
for(i=0;i<v[temp].size();i++)
{
if(dist[v[temp][i].v]>dist[temp]+v[temp][i].w)
{
dist[v[temp][i].v]=dist[temp]+v[temp][i].w;
if(!visited[v[temp][i].v])
{
visited[v[temp][i].v]=1;
enterCount[v[temp][i].v]++;
if(enterCount[v[temp][i].v]>=n)
return true;
q.push(v[temp][i].v);
}
}
}
}
return false;
}
int main()
{
//ifstream cin("1.txt");
int f,i;
cin>>f;
while(f--)
{
cin>>n>>m>>w;
edgeCount=0;
for(i=1;i<=n;i++)
v[i].clear();
for(i=0;i<m;i++)
{
int s,e,t;
cin>>s>>e>>t;
addEdge(s,e,t);
addEdge(e,s,t);
}
for(i=0;i<w;i++)
{
int s,e,t;
cin>>s>>e>>t;
addEdge(s,e,-t);
}
if(spfa())
cout<<"YES"<<endl;
else
cout<<"NO"<<endl;
}
return 0;
}