【问题描述】
某城市中有一些消防队。居民抱怨说一些房子离最近的消防站很远,所以必须建一个新的消防站。你要选择这个消防站的位置,让离最缺少服务的居民区最近。
这个城市有N个路口,用1..N编号,这N个路口被M条长度不同的双向道路连接。房子和消防站的位置都在路口上。假定在每个路口至少有一栋房子,且在同一个路口可以有多个消防站。
【输入格式】
第一行包含三个整数:f,N,M,分别表示城市现有f个消防站,N个路口和M条道路。
接下来的f行,每行一个整数,表示现有消防站所在路口的编号。接下来的M行,每行包含3个整数,分别表示每条双向道路关联的两个路口编号和这个路口的长度(长度<=1000)。
【输出格式】
包含两行,第一行输出新消防站应建立的路口编号,若有多个,则输出编号最小的一个,第二行,表示新消防站建立后,任意路口的房子到最近消防站的距离的最大值。
【输入样例】
1 6 6
2
1 2 10
2 3 10
3 4 10
4 5 10
5 6 10
6 1 10
【输出样例】
5
10
【数据范围】
f<=500
N<=1000
M<=2000
此题要先进行多源spfa(跟多源bfs是一样的,详见http://blog.csdn.net/qq_35546274/article/details/51896423)。
然后再枚举每一个可以新建的地方(已经有的就不用枚举了,因为再建一个不可能是最优情况),然后选每种情况的最小的最大边,进行判断。
详见代码:
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<vector>
#include<queue>
using namespace std;
const int maxn=1005;
const int inf=200000000;
vector<int>g[maxn],w[maxn];
int n,m,s,vis[maxn]={0},d[maxn],d2[maxn],a[maxn]={0};
queue<int>q;
void init()
{
scanf("%d%d%d",&s,&n,&m);
for(int i=1;i<=n;i++)
d[i]=inf;
for(int i=1;i<=s;i++)
{
int x;
scanf("%d",&x);
q.push(x);
d[x]=0;
vis[x]=1;
a[x]=1;
}
for(int i=1;i<=m;i++)
{
int x,y,z;
scanf("%d%d%d",&x,&y,&z);
g[x].push_back(y);
g[y].push_back(x);
w[x].push_back(z);
w[y].push_back(z);
}
}
void spfa()
{
while(!q.empty())
{
int i=q.front();
q.pop();
vis[i]=0;
for(int k=0;k<g[i].size();k++)
{
int j=g[i][k],c=w[i][k];
if(d[i]+c>=d[j]) continue;
d[j]=d[i]+c;
if(vis[j]) continue;
q.push(j);
vis[j]=1;
}
}
}
void spfa2(int x)
{
queue<int>q;
q.push(x);
d2[x]=0;
vis[x]=1;
while(!q.empty())
{
int i=q.front();
q.pop();
vis[i]=0;
for(int k=0;k<g[i].size();k++)
{
int j=g[i][k],c=w[i][k];
if(d2[i]+c>=d2[j]) continue;
d2[j]=d2[i]+c;
if(vis[j]) continue;
q.push(j);
vis[j]=1;
}
}
}
void in()
{
for(int i=1;i<=n;i++)
d2[i]=inf;
}
int main()
{
init();
spfa();
int ans=inf,k;
for(int i=1;i<=n;i++) if(a[i]==0)
{
in();
spfa2(i);
int t=0;
for(int j=1;j<=n;j++)
{
t=max(min(d[j],d2[j]),t);
}
if(t<ans)
{
ans=t;
k=i;
}
}
printf("%d\n%d",k,ans);
return 0;
}