ccf_201403-4_无线网络

试题编号： 201403-4
试题名称： 无线网络
时间限制： 1.0s
内存限制： 256.0MB
问题描述
　　目前在一个很大的平面房间里有 n 个无线路由器,每个无线路由器都固定在某个点上。任何两个无线路由器只要距离不超过 r 就能互相建立网络连接。
　　除此以外,另有 m 个可以摆放无线路由器的位置。你可以在这些位置中选择至多 k 个增设新的路由器。
　　你的目标是使得第 1 个路由器和第 2 个路由器之间的网络连接经过尽量少的中转路由器。请问在最优方案下中转路由器的最少个数是多少?
输入格式
　　第一行包含四个正整数 n,m,k,r。(2 ≤ n ≤ 100,1 ≤ k ≤ m ≤ 100, 1 ≤ r ≤ 108)。
　　接下来 n 行,每行包含两个整数 xi 和 yi,表示一个已经放置好的无线 路由器在 (xi, yi) 点处。输入数据保证第 1 和第 2 个路由器在仅有这 n 个路由器的情况下已经可以互相连接(经过一系列的中转路由器)。
　　接下来 m 行,每行包含两个整数 xi 和 yi,表示 (xi, yi) 点处可以增设 一个路由器。
　　输入中所有的坐标的绝对值不超过 108,保证输入中的坐标各不相同。
输出格式
　　输出只有一个数,即在指定的位置中增设 k 个路由器后,从第 1 个路 由器到第 2 个路由器最少经过的中转路由器的个数。
样例输入
5 3 1 3
0 0
5 5
0 3
0 5
3 5
3 3
4 4
3 0
样例输出
2

#include <iostream>
#include <vector>
#include <queue>
using namespace std;
class Router{
    long long x,y;//坐标
public:
    bool vis;//是否已遍历
    int d,k,number;//层数，经过的新路由数，编号
    Router() = default;//默认构造函数
    Router(long long x,long long y,int number):x(x),y(y),number(number),vis(false),d(-1),k(-1){}//构造函数
    bool Connect(Router item,long long r){//判断是否可连接
        return (x-item.x)*(x-item.x) + (y-item.y)*(y-item.y) <= r*r;
    }
};
int main(){
    int m,n,k;
    long long r;
    //读取数据
    cin>>n>>m>>k>>r;
    vector<Router> routers;
    for (int i = 1; i <= n + m; ++i) {
        long long x,y;
        cin>>x>>y;
        routers.emplace_back(x,y,i);
    }
    //广度优先搜索（BSF）
    queue<Router> que;
    routers[0].vis = true;
    routers[0].d = 0;
    routers[0].k = 0;
    que.push(routers[0]);
    while(!que.empty()){
        Router u = que.front();
        que.pop();
        if(u.number == 2) break;//到达目标，退出搜索
        for (int i = 0; i < (u.k == k?n:n+m); ++i){//如果当前节点的k已等于新设路由个数，则不再搜索后m个路由
            Router &v = routers[i];
            if((!v.vis) && u.Connect(v,r)){
                v.vis = true;
                v.d = u.d + 1;
                if(v.number > n)//当前节点编号大于n，则当前节点的k等于父节点的k加1，否则，继承父节点的k值
                    v.k = u.k + 1;
                else v.k = u.k;
                que.push(v);
            }
        }
    }
    cout<<routers[1].d-1;
}

广度优先搜索算法