`
Midnight0101
  • 浏览: 15878 次
  • 性别: Icon_minigender_1
  • 来自: 天津
最近访客 更多访客>>
社区版块
存档分类
最新评论

poj3259 spfa解法

poj 
阅读更多
同上题,不过改的spfa算法,注意每个节点进入队列的次数至多为n-1次(一共n个节点),若进入大于等于n次了,则说明图中存在负权回路,此时正好满足题目中时光倒流的要求,另外注意,vector每次用的时候清空。
下面是spfa算法的简单说明:
我们用数组d记录每个结点的最短路径估计值,而且用邻接表来存储图G。我们采取的方法是松弛:设立一个先进先出的队列用来保存待优化的结点,优化时每次取出队首结点u,并且用u点当前的最短路径估计值对离开u点所指向的结点v进行松弛操作,如果v点的最短路径估计值有所调整,且v点不在当前的队列中,就将v点放入队尾。这样不断从队列中取出结点来进行松弛操作,直至队列空为止。
注意对刚出队列的节点的所有相邻节点都要做松弛操作,不管该节点是否在队列中,不在队列中的松弛点加入到队列即可。
#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;
}

分享到:
评论

相关推荐

Global site tag (gtag.js) - Google Analytics