CCF CSP 竞赛试题——无线网络（201403-4）

问题描述

　　目前在一个很大的平面房间里有 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 <utility>
#include <vector>
#include <algorithm>
#include <queue>
using namespace std; 

int N;
int M;
int K;
int R;
vector<pair<int, int> > sites;

bool isConnected(const pair<int, int> &a, const pair<int, int> &b) {
    int w = a.first-b.first;
    int h = a.second-b.second;
    return (long long)R * R >= (long long)w*w + (long long)h*h;
}

int main() {
    cin >> N >> M >> K >> R;
    int x, y;
    for (int i = 0; i < N+M; i++) {
        cin >> x >> y;
        sites.push_back(pair<int, int>(x, y));
    }
    vector<bool> used(N+M, false);
    vector<int> newSitesUsed(N+M, 0);
    queue<int> myque;
    used[0] = true;
    myque.push(0);
    int numOfSites = 0; 
    bool flag = true;
    while (flag) {
        int sz = myque.size();
        int cur;
        for (int i = 0; i < sz; i++) {
            cur = myque.front();
            myque.pop();
            if (cur == 1) {
                flag = false;
                break;
            }
            for (int j = 0; j < N+M; j++) {
                if (j != cur && isConnected(sites[j], sites[cur])) {
                    if (!used[j]) {
                        int t = (j >= N ? newSitesUsed[cur]+1 : newSitesUsed[cur]);
                        if (t <= K) {
                            used[j] = true;
                            newSitesUsed[j] = t;
                            myque.push(j);
                        }
                    } else {
                        int t = (j >= N ? newSitesUsed[cur]+1 : newSitesUsed[cur]);
                        if (t < newSitesUsed[j]) {
                            newSitesUsed[j] = t;
                            myque.push(j);
                        }
                    }
                }
            }
        }
        numOfSites++;
    }
    cout << numOfSites-2 << endl;
    return 0;
}