这是一道最短路的变形题,在一个平面图上,两个点之间只要距离不大于r,就可以连通
所以一开始就先计算所有点(包括需要新增的点)之间的距离,如果不大于r,就添加无向边连接这两个点
在最短路搜索时,不仅要记录路劲长度,还要记录路过的新增点的数量(不能大于K),用迪杰斯特拉算法
为了保证答案正确,不陷入局部最优解,需要将一般的dist[N],改变为dist[N][M](N为点,M为路过的新增点个数)
最后在dist[1][0]~dist[1][K]中取最小值作为答案
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
const int N=200+10; //路由器总数量
const int M=100+10;
int first[N];//初始化为-1
int dist[N][M]; //最短距离,前一个代表是哪个点,后一个代表经过了多少个新增路由器
struct Edge
{
int to,v,next;
}edge1[2*N*N]; //邻接表
int tt; //初始化为0
void add(int a,int b) //添加边
{
edge1[tt].to=b;
edge1[tt].next=first[a];
first[a]=tt++;
}
struct node
{
int p,d,k; //位置,距离,经过的新增路由器数量
bool operator < (const node &u) const
{
return d>u.d;
}
node(int a,int b,int c)
{
p=a;d=b;k=c;
}
};
int n,m,k;
double r,x[N],y[N]; //坐标
int main()
{
scanf("%d%d%d%lf",&n,&m,&k,&r);
int cnt=0;
memset(first,-1,sizeof(first));
for(int i=0;i<n+m;i++)
{
scanf("%lf%lf",&x[cnt],&y[cnt]);
cnt++;
for(int j=0;j<cnt-1;j++) //检查两两之间能否连通,可以的话直接加边
{
double d=sqrt((x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j]));
if(d<=r)
{
add(i,j);
add(j,i);
}
}
}
priority_queue<node> q; //迪杰斯特拉
memset(dist,999999,sizeof(dist));
q.push(node(0,0,0));
memset(dist[0],0,sizeof(dist[0]));
while(!q.empty())
{
node u=q.top();
q.pop();
for(int i=first[u.p];i!=-1;i=edge1[i].next)
{
int v=edge1[i].to;
int vd=u.d+1,vk=u.k;
if(v>=n) vk++; //这个点是新增路由器,k+1
if(vk>k) continue; //不能再新增了
if(dist[v][vk]>vd) //松弛
{
dist[v][vk]=vd;
q.push(node(v,vd,vk));
}
}
}
int ans=999999;
for(int i=0;i<M;i++) //找到最短的路
ans=min(ans,dist[1][i]);
printf("%d\n",ans-1);
return 0;
}