ACM P服务点设置(枚举DFS水题)

5. P服务点设置

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

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

 

【输入格式】

输入由若干行组成，第一行有3个整数，n（1≤n≤100）、m（1≤m≤n*n）,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

NP难，直接枚举dfs

#include <iostream>
#include <cstring>
#include <cstdio>

using namespace std;

#define MAX_N 100
#define INF 99999999

int n,m,p;
int d[MAX_N][MAX_N];
int cd[MAX_N][MAX_N];

int minz=INF;
int sel[MAX_N];
int minsel[MAX_N];
bool used[MAX_N];

void dfs(int cur)
{
	if(cur==p)
	{
		int maxd=0;
		for(int i=0;i<n;i++)
		{
			int mind=INF;
			for(int j=0;j<p;j++)
			{
				int s=sel[j];
				mind=min(d[s][i],mind);
			}
			maxd=max(maxd,mind);
		}

		if(minz>maxd)
		{
			minz=maxd;
			for(int i=0;i<p;i++) minsel[i]=sel[i];
		}

		return;
	}

	for(int i=0;i<n;i++)
	{
		if(!used[i])
		{
			sel[cur]=i;
			used[i]=true;
			dfs(cur+1);
			used[i]=false;
		}
	}
}

int main()
{
	freopen("djsc.in","r",stdin);
	freopen("djsc.out","w",stdout);

	cin>>n>>m>>p;
	for(int i=0;i<n;i++)
	{
		for(int j=0;j<n;j++)
		{
			if(i!=j)
			d[i][j]=INF;
			else
			d[i][j]=0;
		}
	}

	for(int i=0;i<m;i++)
	{
		int b,e,v;
		cin>>b>>e>>v;
		d[e][b]=d[b][e]=min(v,d[b][e]);
	}

	for(int k=0;k<n;k++)
	{
		for(int i=0;i<n;i++)
		{
			for(int j=0;j<n;j++)
				d[i][j]=min(d[i][j],d[i][k]+d[k][j]);
		}
	}
	memset(used,0,sizeof(used));
	dfs(0);

	for(int i=0;i<p-1;i++) cout<<minsel[i]<<" ";
	cout<<minsel[p-1]<<endl;

	return 0;
}