COGS 5. P服务点设置

5. P服务点设置

http://www.cogs.pro/cogs/problem/problem.php?pid=5

★★   输入文件：djsc.in   输出文件：djsc.out   简单对比
时间限制：2 s   内存限制：128 MB

 问题描述
为了进一步普及九年义务教育，政府要在某乡镇建立P所希望小学，该乡镇共有n个村庄，村庄间的距离已知，请问学校建在哪P个村庄最好？（好坏的标准是学生就近入学，即在来上学的学生中，以最远的学生走的路程为标准。或者说最远的学生与学校的距离尽可能的小。）

 

【输入格式】

输入由若干行组成，第一行有3个整数，n（1≤n≤100）、m（1≤m≤n2）,p；n表示村庄数，m表示村庄间道路数。第2至m+1行是每条路的信息，每行三个整数，为道路的起点、终点和两村庄间距离。（村庄从0开始编号）

【输出格式】

P个整数，学校所在村庄编号(如果P个以上村庄都适合建立学校，选择编号小的P个村庄建学校，输出时按编号从小到大输出)。

【输入样例】

输入文件名：djsc.in

6 8 2
0 2 10
0 4 30
0 5 100
1 2 5
2 3 50
3 5 10
4 3 20
4 5 60

【输出样例】

输出文件名：djsc.out

0 3

floyd + 递归枚举每个学校的位置

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int n,m,p,ans=0x7fffffff;
int f[101][101];
int b[101],c[101];
void dfs(int sum,int now)
{
    if(sum==p+1)
    {
        int maxn=0;
        for(int i=0;i<n;i++)
        {
            int minn=0x7fffffff;
            for(int j=1;j<=p;j++)
             minn=min(minn,f[i][b[j]]);
            maxn=max(maxn,minn);
        }
        if(maxn<ans)
        {
            ans=maxn;
            memset(c,0,sizeof(c));
            for(int i=0;i<n;i++) c[i]=b[i];
        }
        return;
    }
    for(int i=now+1;i<n;i++)
     {
          b[sum]=i;
          dfs(sum+1,i);
     }
}
int main()
{
    freopen("djsc.in","r",stdin);
    freopen("djsc.out","w",stdout);
    scanf("%d%d%d",&n,&m,&p);
    for(int i=0;i<n;i++)
     for(int j=0;j<n;j++) 
      f[i][j]=1e9/2;
    int u,v,w;
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&u,&v,&w);
        f[u][v]=f[v][u]=w;
    }
    for(int i=0;i<n;i++) f[i][i]=0;
    for(int i=0;i<n;i++)
     for(int j=0;j<=n;j++)
      for(int k=0;k<n;k++) 
       f[j][k]=min(f[j][k],f[j][i]+f[i][k]);
    dfs(1,-1);
    for(int i=1;i<=p;i++) printf("%d ",c[i]);
}