POJ - 3268 Silver Cow Party 最短路，反向建图

题目链接:

POJ - 3268 .

题意：

给定n个节点，m条带权单向路，特殊节点x，求所有节点到达x再返回的距离最大值。

解题思路：

n范围1k，多源的floyd会超时，考虑dij或者spfa。首先从x返回各个节点的距离就是dij跑出来的dis数组，问题在于计算各个节点前往x的距离，因为是单向路，这个距离和dij跑出的dis是不一样的。所以考虑反向建图，正向图反向图分别跑一遍dij算法，将两个dis数组求和，枚举找到最大值输出即可。

代码：

#include<iostream>
#include<cstring>
#include<queue>  
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define endl "\n"
using namespace std;
	typedef long long ll;
	typedef pair <int,int> P;
	const int maxn=1005;
	const int maxe=100000;
	const int inf=0x3f3f3f3f;
	int head1[maxn],head2[maxn];
	struct Edge{
		int to;
		int next;
		int w;
	} edge1[maxe],edge2[maxe];//两张图
	int cnt;
	int dis2[maxn];
 	int dis[maxn];	
	void init(){
		cnt=0;
		memset(head1,-1,sizeof(head1));
		memset(head2,-1,sizeof(head2));
		return ;
	}
	inline void add(int u,int v,int w){//正向图正向边，反向图反向边
		edge1[cnt].next=head1[u];
		edge1[cnt].to=v;
		edge1[cnt].w=w;
		head1[u]=cnt;
		
		edge2[cnt].next=head2[v];
		edge2[cnt].to=u;
		edge2[cnt].w=w;
		head2[v]=cnt;
		cnt++;
	}
			
	void dij1(int start){//正向图dij
		memset(dis,0x3f,sizeof(dis));
		priority_queue<P,vector<P>,greater<P> > q;
		dis[start]=0;
		q.push(P(0,start));
		while(!q.empty()){
			P p=q.top(); q.pop();
			int v=p.second;
			
			if(dis[v]<p.first)		continue;
			
			for(int i=head1[v];i!=-1;i=edge1[i].next){
				int tmp=edge1[i].to;
				if(dis[tmp]>dis[v]+edge1[i].w){
					dis[tmp]=dis[v]+edge1[i].w;
					q.push(P(dis[tmp],tmp));
				}
			}
		}
		return ;
	}
	void dij2(int start){//反向图dij
		memset(dis2,0x3f,sizeof(dis2));
		priority_queue<P,vector<P>,greater<P> > q;
		dis2[start]=0;
		q.push(P(0,start));
		while(!q.empty()){
			P p=q.top(); q.pop();
			int v=p.second;
			
			if(dis2[v]<p.first)		continue;
			
			for(int i=head2[v];i!=-1;i=edge2[i].next){
				int tmp=edge2[i].to;
				if(dis2[tmp]>dis2[v]+edge2[i].w){
					dis2[tmp]=dis2[v]+edge2[i].w;
					q.push(P(dis2[tmp],tmp));
				}
			}
		}
		return ;
	}
	int n,m,x;
	int main(){
		IOS
		init();
		cin>>n>>m>>x;
		while(m--){
			int u,v,w;
			cin>>u>>v>>w;
			add(u,v,w);
		}
		dij1(x),dij2(x);
		int ans=-1;
		for(int i=1;i<=n;i++)
			if(dis[i]!=inf&&dis2[i]!=inf&&dis[i]+dis2[i]>ans)
				ans=dis[i]+dis2[i];
		cout<<ans<<endl;
		return 0;
	}