一、题目
有 n 个城市,按从 0 到 n-1 编号。给你一个边数组 edges,其中 edges[i] = [fromi, toi, weighti] 代表 from i和 to i两个城市之间的双向加权边,距离阈值是一个整数 distanceThreshold。
返回能通过某些路径到达其他城市数目最少、且路径距离最大为distanceThreshold的城市。如果有多个这样的城市,则返回编号最大的城市。
注意,连接城市 i 和 j 的路径的距离等于沿该路径的所有边的权重之和。
二、思路
主要用到多源最短路径的知识,Dijkstra适用于单源一对一的最短路径,Floyd适用于多源最短路径问题,直接套用公式即可
主要还要理清题目题干:能通货某些路径到达其他城市的数目最少,矩阵对应行中不为MAX_LEN的数量就是当前城市i可以到达其他城市的数量。路径距离最大表示含义为当前城市到达其他城市的距离要小于等于给定的dis
三、代码
代码如下(示例):
package leetcode.editor.cn;
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;
/**
* 阈值距离内邻居最少的城市
* @author CJ
* @date 2022-07-23 13:27:19
*/
class P1334_FindTheCityWithTheSmallestNumberOfNeighborsAtAThresholdDistance{
public static void main(String[] args) {
//测试代码
Solution solution = new P1334_FindTheCityWithTheSmallestNumberOfNeighborsAtAThresholdDistance().new Solution();
System.out.println("please input n:");
Scanner sc=new Scanner(System.in);
int n=sc.nextInt();
System.out.println("please input edges length:");
int len=sc.nextInt();
int[][] e=new int[len][3];
for(int i=0;i<len;i++)
for(int j=0;j<3;j++)
e[i][j]=sc.nextInt();
System.out.println("please input dis:");
int dis=sc.nextInt();
int res=solution.findTheCity(n,e,dis);
System.out.println(res);
}
//力扣代码
//leetcode submit region begin(Prohibit modification and deletion)
class Solution {
public int findTheCity(int n, int[][] edges, int distanceThreshold) {
// 定义邻接矩阵D,并初始化各个城市间距离为INT_MAX(无穷)
int[][] D = new int[n][n];
for (int i = 0; i < n; i++)
for (int j = 0; j < n; j++)
D[i][j] = Integer.MAX_VALUE;
// 根据edges[][]初始化D[][]
for (int i = 0; i < edges.length; i++) {
D[edges[i][0]][edges[i][1]] = edges[i][2];
D[edges[i][1]][edges[i][0]] = edges[i][2];
}
// Floyd算法
for (int k = 0; k < n; k++) {
// n个顶点依次作为插入点
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++) {
if (i == j || D[i][k] == Integer.MAX_VALUE || D[k][j] == Integer.MAX_VALUE) {
// 这些情况都不符合下一行的if条件,
// 单独拿出来只是为了防止两个INT_MAX相加导致溢出
continue;
}
D[i][j] = Math.min(D[i][k] + D[k][j], D[i][j]);
}
}
}
// 选择出能到达其它城市最少的城市ret
int ret=0;
int minNum = Integer.MAX_VALUE;
for (int i = 0; i < n; i++) {
int cnt = 0;
for (int j = 0; j < n; j++) {
if (i != j && D[i][j] <= distanceThreshold) {
cnt++;
}
}
if (cnt <= minNum) {
minNum = cnt;
ret = Math.max(i,ret);
}
}
return ret;
}
}
//leetcode submit region end(Prohibit modification and deletion)
}
四、总结
要理解题干,本题属于多源最短路径的题目,可以使用Dijkstra和Floyd两种算法,但是Dijkstra对于多源路径的题目来说,体量比较大,容易超时,所以选择使用Floyd,还要注意的问题在于题目中对于返回值的理解。