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

来源：互联网发布：矩阵的叉积怎么算的编辑：程序博客网时间：2024/06/02 07:29

题意：

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

0 0