【问题描述】
某城市中有一些消防队。居民抱怨说一些房子离最近的消防站很远,所以必须建一个新的消防站。你要选择这个消防站的位置,让离最缺少服务的居民区最近。
这个城市有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,先枚举一遍路口i,如果这个路口没有消防站,就设它有一个消防站,然后再和输入数据的消防站一起压入队列进行多源SPFA,找出距离消防站最近的房子的路径长度,再和其它的最近长度值中比较一下找到最大值。
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<queue>
#include<vector>
using namespace std;
const int maxn=1002,inf=1000000000;
int f,n,m,t,a,b,c,fi[maxn],inq[maxn],d[maxn];
vector<int>g[maxn],w[maxn];
queue<int>q;
void spfa(int s)
{
queue<int>q;
memset(inq,0,sizeof(inq));
for(int i=1;i<=n;i++)
d[i]=inf;
for(int i=1;i<=n;i++) //准备多源SPFA
if(fi[i])
{
d[i]=0;
inq[i]=1;
q.push(i);
}
while(!q.empty())
{
int i=q.front(); q.pop();
inq[i]=0;
for(int k=0;k<g[i].size();k++)
{
int j=g[i][k];
if(d[i]+w[i][k]<d[j])
{
d[j]=d[i]+w[i][k];
if(inq[j]) continue;
inq[j]=1;
q.push(j);
}
}
}
}
int main()
{
//freopen("1.txt","r",stdin);
scanf("%d%d%d",&f,&n,&m);
memset(fi,0,sizeof(fi));
for(int i=1;i<=f;i++)
{
scanf("%d",&t);
fi[t]++;
}
for(int i=1;i<=m;i++)
{
scanf("%d%d%d",&a,&b,&c);
g[a].push_back(b);
g[b].push_back(a);
w[a].push_back(c);
w[b].push_back(c);
}
int ans=inf,id=0;
for(int i=1;i<=n;i++) //枚举一遍
{
if(!fi[i]) //如果这里没有消防站
{
fi[i]=1; //假设一个消防站
spfa(i);
int t=0;
for(int j=1;j<=n;j++) //找距离消防站最近的房子的路径长度
if(!fi[j])
t=max(t,d[j]);
if(ans>t) ans=t,id=i;
fi[i]=0; //复原
}
}
printf("%d\n%d",id,ans);
return 0;
}