回旋镖的数量的思路探讨与源码
回旋镖的数量的题目如下图,该题属于数组类型和哈希表类型的题目,主要考察对于数学方法的使用和数组结构的理解。本文的题目作者想到2种方法,分别是哈希表方法和枚举方法,其中哈希表方法使用Java进行编写,而枚举方法使用Python进行编写,当然这可能不是最优的解法,还希望各位大佬给出更快的算法。
本人认为该题目可以使用哈希表方法的思路进行解决,首先计算数组的长度并初始化参数,然后开始循环遍历,在每次遍历开始都初始化一个哈希表,再进行内部的子循环,如果下标相等就结束本次子循环,如果否就计算下标对应的数组的值的距离,并计算横竖坐标的距离的平方和,并把结果放入哈希表中。遍历子循环结束以后的哈希表,获取距离值对应的结果并加入到最终结果里面,直到遍历结束并返回最终结果。那么按照这个思路我们的Java代码如下:
#喷火龙与水箭龟
class Solution {
public int numberOfBoomerangs(int[][] points) {
int numLen = points.length;
int res = 0;
for (int ir = 0; ir < numLen; ir++) {
Map<Integer, Integer> resMap = new HashMap<>();
for (int jr = 0; jr < numLen; jr++) {
if (ir == jr){
continue;
}
int x = points[ir][0] - points[jr][0];
int y = points[ir][1] - points[jr][1];
int dst = x * x + y * y;
resMap.put(dst, resMap.getOrDefault(dst, 0) + 1);
}
for (int dst : resMap.keySet()) {
int countNum = resMap.get(dst);
res = res + countNum * (countNum - 1);
}
}
return res;
}
}
显然,我们的哈希表方法的效果一般,还可以用枚举方法进行解决。首先初始化参数,开始遍历数组,在每次遍历开始都初始化一个字典,开始子循环遍历,计算距离和并把结果放入到字典里,直到子循环遍历结束后,把字典结果全部取出并按规律计算结果后累加,直到全部遍历结束后返回最终计算结果。所以按照这个思路就可以解决,下面是Python代码:
#喷火龙与水箭龟
class Solution:
def numberOfBoomerangs(self, points: List[List[int]]) -> int:
res = 0
for p in points:
resMap = defaultdict(int)
for q in points:
des = (p[0] - q[0]) * (p[0] - q[0]) + (p[1] - q[1]) * (p[1] - q[1])
resMap[des] = resMap[des] + 1
for v in resMap.values():
res = res + v * (v - 1)
return res
从结果来说Java版本的哈希表方法的效率一般,而Python版本的枚举方法的速度比较不错,但应该是有更多的方法可以进一步提速的,希望朋友们能够多多指教,非常感谢。