【CCF-CSP】无线网络

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

5 3 1 3
0 0
5 5
0 3
0 5
3 5
3 3
4 4
3 0

样例输出

2

算法

通过分析题目我们得知需要求的是所有从1到2经过的特殊点数小于等于k的最短路径。
则转化为利用宽度优先搜索(bfs)求单源最短路经的问题。

#include <iostream>
#include <cstring>
#include <algorithm>
#include <queue>

#define x first
#define y second

using namespace std;

typedef pair<int, int> PII;
typedef long long LL;

const int N = 210, M = N * N;

int n, m, k, r;
int h[N], e[M], ne[M], idx;
PII p[N]; //用于存储点集
int dist[N][N]; //存储距离

bool check(PII a, PII b) {
    LL dx = a.x - b.x;
    LL dy = a.y - b.y;
    return dx * dx + dy * dy <= (LL)r * r;
}

void add(int a, int b) {
    e[idx] = b, ne[idx] = h[a], h[a] = idx ++;
}

int bfs() {
    queue<PII> q;
    //最开始在1号点,没有经过任何一个特殊点
    q.push({1, 0});
    memset(dist, 0x3f, sizeof dist);
    dist[1][0] = 0;
    
    while (q.size()) {
        auto t = q.front();
        q.pop();
        
        //枚举从t点出发能够到达的所有边
        for (int i = h[t.x]; ~i; i = ne[i]) {
            int x = e[i]; //这条边指向点的编号
            int y = t.y; //y表示走过的特殊点的数量
            //若为特殊点,则走过的特殊点数量+1
            if (x > n) y ++;
            //不能用超过k个点
            if (y <= k) {
                if (dist[x][y] > dist[t.x][t.y] + 1) {
                    dist[x][y] = dist[t.x][t.y] + 1;
                    q.push({x, y});
                }
            }
        }
    }
    
    int res = 1e8;
    for (int i = 0; i <= k; i ++ )
        res = min(res, dist[2][i]);
    return res - 1;
}

int main() {
    cin >> n >> m >> k >> r;
    memset(h, -1, sizeof h);
    for (int i = 1; i <= n; i ++ ) cin >> p[i].x >> p[i].y;
    for (int i = n + 1; i <= n + m; i ++ ) cin >> p[i].x >> p[i].y;
    
    //对点集插入边
    for (int i = 1; i <= n + m; i ++ )
        for (int j = i + 1; j <= n + m; j ++ )
            //两个无线路由器距离不超过r就能互相建立网络连接
            if (check(p[i], p[j]))
                add(i, j), add(j, i);
                
    cout << bfs() << endl;
    return 0;
}
上一篇:2021 CSP-J 第一轮总结/游记


下一篇:CSP-S 2021游记