CCF201403-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<queue>
#include<cstring>
#include<cmath>
#include<fstream>
#define N 210
#define INF 100000001
typedef long long LL ;
using namespace std;

//无线路由器：横纵坐标 
struct R{
    int x;
    int y;
};

struct P_D{
    int dest;       //目的结点 
    int add;        //通过的增设结点的数量 
};

int n;      //路由器个数
int m;      //可摆放路由器的位置
int k;      //可新增的路由器个数
int r;      //可建立连接的距离
R ruter[N];         //已有的和可增设的路由器 
int d[N][N];        //d[i][j]表示从起点开始经过增设的j个路由器到达i的最短路径
bool vis[N][N];     //vis[i][j]表示是否可以从起点经过增设的j个路由器到达i 
bool Map[N][N];     //邻接矩阵 

//  使用spfa算法求最短路径
void spfa(){
    queue <P_D> Q;      //保存结点的队列
    P_D s, temp;        //源点，目的结点 
    s.dest = 0;
    s.add = 0;
    Q.push( s );        //初始时队列中只有起始点

    memset( d, INF, sizeof(d) );
    memset( vis, 0, sizeof(vis) );
    d[0][0] = 0;
    vis[0][0] = 1; 

    while( !Q.empty() ){
        s = Q.front();
        Q.pop();
        vis[s.dest][s.add] = 0;

        for( int i = 0; i < n+m; i++ ){
            if( Map[s.dest][i] ){
                temp.dest = i;
                temp.add = s.add;
                if( i >= n ){
                    temp.add++;
                }
                if( temp.add <= k && d[temp.dest][temp.add] > d[s.dest][s.add]+1 ){
                    d[temp.dest][temp.add] = d[s.dest][s.add]+1;
                    if( !vis[temp.dest][temp.add] ){
                        vis[temp.dest][temp.add] = 1;
                        Q.push( temp );
                    }
                }
            }
        } 
    } 

    int count = INF;
    for( int i = 0; i <= k; i++ ){
        count = min( count, d[1][i] );
    }
    cout << count - 1;
} 

int main(){ 
//  读入数据 
//  ifstream fin("00.txt", ios::in);
//  fin >> n >> m >> k >> r;    
    cin >> n >> m >> k >> r;    
    for( int i = 0; i < n+m; i++ ){
//      fin >> ruter[i].x >> ruter[i].y; 
        cin >> ruter[i].x >> ruter[i].y; 
    }

//  初始化无向图邻接矩阵，无边置0 ，有边置1
    memset( Map, 0, sizeof(Map) );
    for( int i = 0; i < n+m; i++ ){
        for( int j = i+1; j < n+m; j++ ){
            if( (LL)pow((ruter[i].x - ruter[j].x), 2) + (LL)pow((ruter[i].y - ruter[j].y), 2) <= (LL)r*r ){
                Map[i][j] = Map[j][i] = 1; 
            } 
        }
    }

//  使用spfa算法求最短路径
    spfa(); 

    return 0;
}