题外:现在真就只能做做模拟了,菜啊。参考博客:1 2
ps: ccf官网这个题的数据确实水了。
思路:在普通spfa的dis和vis数组上增加一维。dis[i][j]表示源点到i,经过了j个新增中转路由器的距离。(这个距离即一共经过了多少路由器),vis数组含义差不多,只不过是用来标记的。
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
using namespace std;
const int N = 205;
typedef long long ll;
#define inf 0x3f3f3f3f
int n,m,k,r;
struct node
{
int x,y;
}a[N];
int dis[N][N],vis[N][N];
int spfa()
{
queue<node> q;
q.push({0,0});
memset(dis,inf,sizeof dis);
vis[0][0]=1;
dis[0][0]=0;
while(q.size())
{
node tem=q.front();
q.pop();
int x=tem.x,y=tem.y;
vis[x][y]=0;
for(int i=0;i<n+m;i++)
{
if(hypot(a[x].x-a[i].x,a[x].y-a[i].y)<=r*1.0) //可达
{
int t=y+(i>=n?1:0);
if(t<=k&&dis[i][t]>dis[x][y]+1) //距离可更新
{
dis[i][t]=dis[x][y]+1;
if(!vis[i][t])
vis[i][t]=1,q.push({i,t});
}
}
}
}
int ans=inf;
for(int i=0;i<=k;i++)
ans=min(ans,dis[1][i]);
return ans-1;
}
int main()
{
cin>>n>>m>>k>>r;
for(int i=0;i<n+m;i++)
{
int x,y;
cin>>x>>y;
a[i]={x,y};
}
cout<<spfa()<<endl;
return 0;
}