201403-4 无线网络

问题描述

　　目前在一个很大的平面房间里有 n 个无线路由器,每个无线路由器都固定在某个点上。任何两个无线路由器只要距离不超过 r 就能互相建立网络连接。
　　除此以外,另有 m 个可以摆放无线路由器的位置。你可以在这些位置中选择至多 k 个增设新的路由器。
　　你的目标是使得第 1 个路由器和第 2 个路由器之间的网络连接经过尽量少的中转路由器。请问在最优方案下中转路由器的最少个数是多少?

输入格式

　　第一行包含四个正整数 n,m,k,r。(2 ≤ n ≤ 100,1 ≤ k ≤ m ≤ 100, 1 ≤ r ≤ 10⁸)。
　　接下来 n 行,每行包含两个整数 x_i 和 y_i,表示一个已经放置好的无线 路由器在 (x_i, y_i) 点处。输入数据保证第 1 和第 2 个路由器在仅有这 n 个路由器的情况下已经可以互相连接(经过一系列的中转路由器)。
　　接下来 m 行,每行包含两个整数 x_i 和 y_i,表示 (x_i, y_i) 点处可以增设 一个路由器。
　　输入中所有的坐标的绝对值不超过 10⁸,保证输入中的坐标各不相同。

输出格式

　　输出只有一个数,即在指定的位置中增设 k 个路由器后,从第 1 个路 由器到第 2 个路由器最少经过的中转路由器的个数。

样例输入

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

样例输出

2

 

坑题。讲道理，简单的BFS根本行不通的，因为有k的限制。第一次用DFS，搜索了所有的路径并找到添加节点数符合小于给定k值的最小路径，结果超时，只得了20分。后来试了下BFS，不管k的限制竟然满分（如果k给定的小的话，这样得到的路径需要添加的节点大于k那这种情况就不符合，很明显测试数据里的k值都不小）！！！

不废话了，代码:

#include<iostream>
#include<vector>
#include<queue>
using namespace std;

vector<int> vec[205];

int p[205][3];			// 0 x, 1 y , 2 类型 
int n,m,k,r;

void addEdge(int a,int b){
	vec[a].push_back(b);
	vec[b].push_back(a);
}

int countK = 0,minDeep=1000;
int status[205]={0};
int nodeDeep[205]={0};	
/**
	status 0：未遍历，1：正在遍历，2：完成遍历 
**/

int bfs(){

	queue<int> que ;
	que.push(0);
	status[0]=1;
	nodeDeep[0] = 0;
	while(que.size()>0){
		int nowV = que.front();
		que.pop();
		
		status[nowV] = 1;
		
		for(int i=0;i<vec[nowV].size();i++){
			int tmp = vec[nowV][i];
			if(status[tmp]==0){
				status[tmp] = 1;
				nodeDeep[tmp] = nodeDeep[nowV]+1;;
				que.push(tmp);
			}
			if(tmp==1)return nodeDeep[tmp];
		}
		status[nowV] = 2;	
	}
	
}

int main(){
	
	int i,j,l;
	cin>>n>>m>>k>>r;
	for(i=0;i<n+m;i++){
		cin>>p[i][0]>>p[i][1];
		p[i][2] = i>=n?1:0;
	}
	for(i=0;i<m+n-1;i++){
		for(j=i+1;j<m+n;j++){
			long long x2,y2,r2;
			x2 = p[i][0]-p[j][0];	x2*=x2;
			y2 = p[i][1]-p[j][1];	y2*=y2;
			r2 = r;					r2*=r2;
			if(x2+y2<=r2){
				addEdge(i,j);
			}
		}
	}
	cout<<bfs()-1<<endl;
	return 0;
}