[蓝桥杯][历届试题]邮局
时间限制: 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;
}