POJ 2607 Fire Station 消防站

【问题描述】

　　某城市中有一些消防队。居民抱怨说一些房子离最近的消防站很远，所以必须建一个新的消防站。你要选择这个消防站的位置，让离最缺少服务的居民区最近。

　　这个城市有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;
}