201403-4 无线网络 ccf

问题描述

　　目前在一个很大的平面房间里有 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

/*
这里k值貌似没有太大用处，所以其实不记录counts也是可以的，
通过bfs策略得到的一定最优的，所以一定是满足小于等于k的，除非存在不合法例
*/
#include <bits/stdc++.h>
using namespace std;
typedef long long llt;
struct node
{
    int x,y;
    int vertexnum;
    int counts;
    node(int a,int b,int vn,int c):x(a),y(b),vertexnum(vn),counts(c){}
};
vector<node> v;
bool matrix[256][256];
bool visit[256];
bool isConnect(node &a,node &b,int r)//在r范围内
{
    return (llt)(a.x-b.x)*(a.x-b.x)+(llt)(a.y-b.y)*(a.y-b.y)<=(llt)r*r;
}
int main()
{
    int n,m,k,r;
    scanf("%d %d %d %d",&n,&m,&k,&r);
    for(int i=0;i<n+m;++i)
    {
        int x,y;
        scanf("%d %d",&x,&y);
        v.push_back(node(x,y,i,0));
    }
    for(int i=0;i<n+m;++i)//创建邻接矩阵matrix
    {
        for(int j=0;j<n+m;++j)
        {
            matrix[i][j]=isConnect(v[i],v[j],r);
        }
    }
    queue<node> q;
    q.push(v[0]);
    int elecnt=1,level=1;
    while(elecnt>0)//bfs
    {
        int nextelecnt=0;//记录下一层次元素个数
        for(int i=0;i<elecnt;++i)
        {
            node cur=q.front();
            q.pop();
            if(cur.vertexnum==1)//是第二个路由器，得到结果
            {
                printf("%d\n",level-2);
                return 0;
            }
            int w=cur.vertexnum,u=0;
            while(u<m+n)//依次遍历n+m个节点
            {
                if(matrix[w][u]&&!visit[u]&&cur.counts<=k)
                {
                    visit[u]=true;
                    v[u].counts=(u>=n)?cur.counts+1:cur.counts;
                    q.push(v[u]);
                    ++nextelecnt;
                }
                ++u;
            }
        }
        elecnt=nextelecnt;
        ++level;
    }
    return 0;
}