当前位置: 代码迷 >> 综合 >> poj 2607 Fire Station dijkstra算法求到点集的最短路
  详细解决方案

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

热度:57   发布时间:2024-01-19 06:19:46.0

题意:

给一个有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;	
}


  相关解决方案