建图:
两个路由器可以通讯则二者连一条无向边。
用d[i][j]表示从起点开始经过增设的j个路由器到达i的最短路径,vis[i][j]表示是否可以从起点经过增设的j个路由器到达i。
然后就和普通的最短路一样了。这里选用spfa求解。
最后d[1][0]~d[1][k]的最小值就是答案
(http://115.28.138.223/view.page?gpid=T7)
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<queue>
#define N 205
#define INF 0x3f3f3f3f
typedef long long LL;
using namespace std;
struct Point{
int x;
int y;
}p[205];//定义结构体类型的同时也定义了结构体数组
struct Node
{
int des;
int inc;
};/*des表示目的地 inc表示经过新增路由器数目*/
int n,m,k,r;
int d[205][205];
bool vis[205][205],Map[205][205];
void spfa()
{
queue<Node> q;
memset(vis,0,sizeof(vis));
memset(d,INF,sizeof(d));
Node ago,temp;//之前结点和当前结点
ago.des=0;
ago.inc=0;
d[0][0]=0;
vis[0][0]=1;
q.push(ago);//将起点入栈
while(!q.empty())
{
ago=q.front();
q.pop();
vis[ago.des][ago.inc]=0;
for(int i=0;i<n+m;++i){
if(Map[ago.des][i]){
temp.des=i;
temp.inc=ago.inc;
if(i>=n){//如果编号大于等于n 说明是路由器 即经过的新增路由器数目增1
++temp.inc;
}
if(temp.inc<=k&&d[temp.des][temp.inc]>d[ago.des][ago.inc]+1){
d[temp.des][temp.inc]=d[ago.des][ago.inc]+1;
if(!vis[temp.des][temp.inc]){
vis[temp.des][temp.inc]=1;
q.push(temp);//temp的最短路被更新 而且还为未进入队列 则入队
}
}
}
}
}
int ans=INF;
for(int i=0;i<=k;i++)
ans=min(ans,d[1][i]);
printf("%d\n",ans-1);
}
int main()
{
int i,j;
cin>>n>>m>>k>>r;
for(i=0;i<n+m;++i)
scanf("%d%d",&p[i].x,&p[i].y);
memset(Map,0,sizeof(Map));
for(i=0;i<n+m;++i)
{
for(j=i+1;j<n+m;++j){
if(LL(p[i].x-p[j].x)*(p[i].x-p[j].x)+LL(p[i].y-p[j].y)*(p[i].y-p[j].y)<=LL(r)*r){
Map[i][j]=Map[j][i]=1;//距离合适则增加一条无向边
}
}
}
spfa();
return 0;
}