邮局---dfs与剪枝的训练

[蓝桥杯][历届试题]邮局

时间限制: 1Sec 内存限制: 128MB 提交: 219 解决: 60

题目描述
C村住着n户村民，由于交通闭塞，C村的村民只能通过信件与外界交流。为了方便村民们发信，C村打算在C村建设k个邮局，这样每户村民可以去离自己家最近的邮局发信。

现在给出了m个备选的邮局，请从中选出k个来，使得村民到自己家最近的邮局的距离和最小。其中两点之间的距离定义为两点之间的直线距离。

数据规模和约定
对于100%的数据，1< =n< =50，1< =m< =25，1< =k< =10。

输入
输入的第一行包含三个整数n, m, k，分别表示村民的户数、备选的邮局数和要建的邮局数。
接下来n行，每行两个整数x, y，依次表示每户村民家的坐标。
接下来m行，每行包含两个整数x, y，依次表示每个备选邮局的坐标。
在输入中，村民和村民、村民和邮局、邮局和邮局的坐标可能相同，但你应把它们看成不同的村民或邮局。
输出
输出一行，包含k个整数，从小到大依次表示你选择的备选邮局编号。（备选邮局按输入顺序由1到m编号）
样例输入
5 4 2
0 0
2 0
3 1
3 3
1 1
0 1
1 0
2 1
3 2
样例输出
2 4

//邮局 --- 初步设想：dfs+剪枝 
#include <iostream>
#include <vector>
#include <cstring>
#include <cmath>
using namespace std;

const double INF = 1.7e300;
struct node{
	int x,y;    //村民 
}red[100],Post[100];
double dis(node a, node b)
{
	return sqrt((a.x-b.x)*(a.x-b.x) + (a.y-b.y)*(a.y-b.y));
}
int n,m,k;
double mp[101][101];
int flag[100];
double res = INF;    //保存的是最小的距离 
int cur[100];        //保存新生成的邮局的序列 
int ans[100];        //保存已经选择邮局的序列 

void dfs(int st,int sum_k,double sum_r,double tmp_r[])
{   
// 每个参数的含义：st 表明当前邮局的编号， sum_k表示到当前点时已经使用了(sum_k-1)的邮局了
   //sum_r 表示目前已知的到每个村庄的距离和，tmp_r[i]数组表示当前已知的到i个村庄的最短距离。 

	if((k-(sum_k-1))>(m-(st-1))) return; //剪枝1：含义为当前剩余的节点全部用也凑不齐k个邮局
	
	if(st>m && sum_k<=k)  //剪枝2：当邮局超过最大m且必须已经选的数量也在k的范围内
	
	if(sum_k > k+1) return; //剪枝3：大于选的数量一定返回
	
	if(sum_k == k+1) //dfs出口条件 
	{
		if(res > sum_r)
		{  //更新 
			res = sum_r;
			for(int i = 1;i<=k;i++)
			 ans[i] = cur[i]; //更新为现在选的邮局 
		}
		return ;
	 } 
	 
	 double dis_r[55];
	 
	 for(int i = 1;i<=n;i++)
	 {
	 	dis_r[i] = tmp_r[i]; // 
	 }
	 
	 //dfs的0-1问题，即选择当前点或者跳过当前点 
	 dfs(st+1,sum_k,sum_r,dis_r); //跳过当前点看是否能运行下去
	 
	 //否则，试着建立当前点
	 
	 if(flag[st] == 1) return;//剪枝4：当该点被标记时，说明不能用，直接返回
	 
	 cur[sum_k] = st; //选择当前邮局
	 
	 
	 int mark1 = 0, mark2 = 0;
	 if(sum_k == 1)   //选择的第一个邮局
	 {
	 	for(int i =1;i<=n;i++)
	 	{
	 		dis_r[i] = mp[i][st]; //到每个村庄的距离
			 sum_r+=mp[i][st]; 
		 }
		 mark1 = 1; //表明是第一个邮局的标记 
	  } 
	  else
	  {
	  	for(int i = 1;i<=n;i++)
	  	{
	  		if(dis_r[i] > mp[i][st])
	  		{
	  			  mark2 = 1; //表明该点可以起到分担距离减小的作用的标记
				  sum_r += mp[i][st]- dis_r[i];  //这里很重要，这里要减去已经保存的部分然后才能加新的最小值 
				  dis_r[i]=mp[i][st]; 
			  }
		  }
	  }
	  
	  if(mark1==0&&mark2==0) flag[st] = 1;//标记该点不能使用
	  
	   if(mark1 == 1||mark2 == 1) //选择该点往下搜索
	   dfs(st+1,sum_k+1,sum_r,dis_r);
	   
	   return; 
}
int main()
{
	cin>>n>>m>>k;
	double dis_r[55];
    memset(dis_r,-1,sizeof(dis_r));
    
	for(int i = 1;i<=n;i++)
	{
		cin>>red[i].x>>red[i].y;
	}
	for(int i = 1;i<=m;i++)
	{
		cin>>Post[i].x>>Post[i].y;
	}
	for(int i =1;i<=n;i++)
	 {
	 for(int j =1;j<=m;j++)
	 {
	 	mp[i][j]=dis(red[i],Post[j]); 
	 }
	 }
	 
	 dfs(1,1,0,dis_r);
	 
	 for(int i =1;i<=k;i++)
	 {
	 	printf("%d ",ans[i]);
	 }
	 return 0;
}