[Usaco2007 Feb]Cow Party 奶牛派对 dijkstra+堆优化 最短路

题目描述

原题来自：USACO 2007 Feb. Silver

 N(1≤N≤1000)头牛要去参加一场在编号为x(1≤x≤N)的牛的农场举行的派对。有M(1≤M≤100000)条有向道路，每条路长Ti(1≤Ti≤100)；每头牛都必须参加完派对后回到家，每头牛都会选择最短路径。求这N头牛的最短路径（一个来回）中最长的一条的长度。 特别提醒：可能有权值不同的重边。

输入格式

第 1 行： 3个空格分开的整数N,M,X；

第 2...M+1行： 3个空格分开的整数 Ai,Bi,Ti，表示有一条从 Ai到 Bi的路，长度为 Ti。

输出格式

一行一个数，表示最长最短路的长度。

样例

样例输入

4 8 2
1 2 4
1 3 2
1 4 7
2 1 1
2 3 5
3 1 2
3 4 4
4 2 3

样例输出

10

题目的意思就是先求得x到所有点的最短路，在把所有的边的方向反一遍再求最短路，最后求两次最短路的的和的最大值

即

ans=max(dist1[i]+dist2[i],ans) 

代码：

#include 
#define M int(10e5+1)
#define N int(10e3+1)
const int INF=0xfffffff;
#define lop(i,s,e) for(i=s;i<=e;i++)
using namespace  std;
int cnt,head[M],cnt2,head2[M],dis[N];//两个链式向前星一个正一个反
struct Edge
{
	int next;
	int to;
	int w;
}edge[M],edge2[M];
struct Node
{
	int rank;
	int dis;
	friend bool operator <(Node a,Node b){return a.dis < b.dis;}
}node[N];
void add(Edge edge[],int head[],int &cnt,int u,int v,int w)  
{  
	cnt++;
    edge[cnt].w = w;
    edge[cnt].to = v;  
    edge[cnt].next = head[u];  
    head[u] = cnt;  
}
int n,m,x,i,ans;
void djikstra(Edge edge[],int head[])
{
	int vis[N]={0};
	lop(i,1,n)node[i].rank=i,node[i].dis=INF;
	priority_queue q;
	node[x].dis=0;
	q.push(node[x]);
	int current;
	while(!q.empty())
	{
		current=q.top().rank,q.pop();
		if(vis[current])continue;
		for(int i=head[current];i;i=edge[i].next)
		{
			if(node[edge[i].to].dis>node[current].dis+edge[i].w)
			{
				node[edge[i].to].dis=node[current].dis+edge[i].w;
				q.push(node[edge[i].to]);
			}
		}
	}
}
int main(int argc, char const *argv[])
{
	cin>>n>>m>>x;
	int u,v,w;
	lop(i,1,m)cin>>u>>v>>w,add(edge,head,cnt,u,v,w),add(edge2,head2,cnt2,v,u,w);
	djikstra(edge,head);
	lop(i,1,n)
	{
		dis[i]=node[i].dis;
	}
	djikstra(edge2,head2);
	lop(i,1,n)
	{
		ans=max(ans,node[i].dis+dis[i]);
	}
	cout<