poj 2607 Fire Station dijkstra算法求到点集的最短路

题意：

给一个有n个路口的城市，路口之间由路相连，某些路口上建有消防站，现在要在没有建消防站的路口建一个消防站，使没建消防站的路口到建有消防站的路口最短路径的最大值最小。

分析；

枚举没建消防站的点建消防站，然后求所有点到建有消防站点集的最短路。

代码：

//poj 2607
//sepNINE
#include <iostream>
#include <vector>
#include <queue>
using namespace std;
const int maxN=512;
const int maxM=200000;
int head[maxN],fireStation[maxN],dis[maxN],ifSet[maxN];
bool done[maxN];
int e,f,n;
typedef pair<int,int> pii;
priority_queue<pii,vector<pii>,greater<pii> > q;
struct Edge
{
	int v,w,next;
}edge[maxM];

void addedge(int u,int v,int w)
{
	edge[e].v=v;edge[e].w=w;edge[e].next=head[u];
	head[u]=e++;
}


void dij()
{
	while(!q.empty()) q.pop();
	memset(done,0,sizeof(done));
	int i;
	for(i=1;i<=n;++i){
		if(fireStation[i]==0)
			dis[i]=INT_MAX;	
		else{
			dis[i]=0;
			q.push(make_pair(dis[i],i));
		}
	}
	while(!q.empty())
	{
		pii u=q.top();q.pop();
		int x=u.second;
		if(done[x])
			continue;
		done[x]=1;
		for(int i=head[x];i!=-1;i=edge[i].next){
			int v=edge[i].v,w=edge[i].w; 
			if(dis[x]+w<dis[v])
			{
				dis[v]=dis[x]+w;
				q.push(make_pair(dis[v],v));
			}
		} 
	}
}
int trySet(int u)
{
	int i,maxDis;
	fireStation[u]=1;
	maxDis=-1;
	dij(); 
	for(i=1;i<=n;++i)
 		maxDis=max(maxDis,dis[i]);
	fireStation[u]=0;	
	return maxDis;
}

int main()
{
	int i;
	memset(fireStation,0,sizeof(fireStation));
	memset(head,-1,sizeof(head)); 
	scanf("%d%d",&f,&n);
	for(i=1;i<=f;++i){
		int x;
		scanf("%d",&x);
		fireStation[x]=1;
	}
	e=0;
	int u,v,w;
	while(scanf("%d%d%d",&u,&v,&w)==3){
		addedge(u,v,w);
		addedge(v,u,w);
	}
	for(i=1;i<=n;++i)
		ifSet[i]=INT_MAX;
	for(i=1;i<=n;++i)
		if(fireStation[i]==0)
			ifSet[i]=trySet(i);	
	int ans,minx=INT_MAX;
	for(i=1;i<=n;++i)
		if(minx>ifSet[i]){
			minx=ifSet[i];
			ans=i;
		}
	printf("%d",ans);
	return 0;	
}