问题描述
目前在一个很大的平面房间里有 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
一下代码只是考虑了添加一个点的时候,添加多余一个点的情况没有考虑 希望有大佬们可以指点一二。
import java.util.Arrays;
import java.util.Scanner;
public class Wireless {
static Scanner sc;
static int n,m,k,r;
static int[][] edge;
static Apoint[] point;
static boolean[] isVisited;
static int[] dist;
static int[] len;
static final int INF = Integer.MAX_VALUE; // 最大值
public static void main(String[] args) {
sc = new Scanner(System.in);
n=sc.nextInt(); //已经存在的无线网络的点
m=sc.nextInt(); //新增的无线网络的点
k=sc.nextInt(); //表示新增几个点
r=sc.nextInt(); //最短可以联络的距离
point = new Apoint[n+1];
edge = new int[n+2][n+2];
dist = new int[n+2];
len = new int[m];
//存储已经存在的点
for(int i =0;i<n;i++) {
point[i] = new Apoint(i+1,sc.nextInt(),sc.nextInt());
}
//添加一个网络点的情况
for(int k=0;k<m;k++) {
point[n] = new Apoint(n+1,sc.nextInt(),sc.nextInt());
//每次重新初始化edge二维数组
init();
for(int i=0;i<=n;i++)
for(int j=0;j<=n;j++)
if(point[i].num==point[j].num) {
continue;
} else if(((point[i].x-point[j].x)*(point[i].x-point[j].x)+(point[i].y-point[j].y)*(point[i].y-point[j].y))<=9) {
edge[point[i].num][point[j].num]=1;
edge[point[j].num][point[i].num]=1;
}
len[k]=Distras(2);
}
Arrays.sort(len);
//System.out.print(len[0]+" "+len[1]+" "+len[2]);
System.out.println(len[0]-1);
}
public static void init() {
for(int i = 1 ; i <= n+1 ; i++) {
for(int j = 1 ; j <= n+1 ; j++) {
if(i == j)
edge[i][j] = 0;
else
edge[i][j] = INF;
}
}
}
//利用缔结斯塔拉算法 算出单源最短距离
public static int Distras(int end) {
isVisited = new boolean[n+2];
//初始化
for(int i = 1 ; i <=n+1 ; i++) {
isVisited[i] = false; //表明顶点i的最短路径还没有获取到
dist[i] = edge[1][i]; //表明起始点到顶点i的距离
}
//对起始点进行初始化
isVisited[1] = true;
dist[1] = 0;
int k = 0;
//遍历n次 每一次都找到一个顶点的最短路径
for(int i = 0 ; i < n ; i++) {
int min = INF;
for(int j = 1 ; j <= n+1 ; j++) {
if(isVisited[j] == false && min > dist[j]) {
min = dist[j];
k = j;
}
}
isVisited[k] = true;
//更新dist数组
for(int j = 1 ; j <= n+1 ; j++) {
int temp = (edge[k][j] == INF ? INF : (min + edge[k][j]));
if(isVisited[j] == false && (temp < dist[j])) {
dist[j] = temp;
}
}
}
return dist[2];
}
}
class Apoint{
public int num; //点的编号
public int x; //点的x轴的位置
public int y; //点的y轴的位置
public Apoint(int num, int x, int y) {
this.num = num;
this.x = x;
this.y = y;
}
}