题目:
给出一个平面上的n个点,寻找存在多少个由这些点构成的三元组(i, j, k),使得i, j两点的距离等于i, k两点的距离。其中n最多为500,且所有点坐标的范围在[-10000, 10000]之间。
示例:
- 如[[0, 0], [1, 0], [2,0]],则结果为2
- 两个结果为[[1, 0], [0, 0], [2, 0]]和[[1, 0], [2, 0], [0, 0]]
方法一:
观察到i是一个枢纽,对于每个点i,遍历其余点到i的距离。采用查找表的方式。
时间复杂度:O(n^2)
空间复杂度:O(n)
方法一代码:
package com.haobi;
import java.util.HashMap;
import java.util.Map;
public class NumberOfBoomerangs {
public static void main(String[] args) {
int[][]arr = {{0,0},{1,0},{2,0}};
System.out.println(numberOfBoomerangs(arr));
}
public static int numberOfBoomerangs(int[][] points) {
int res = 0;
for(int i=0;i<points.length;i++) {
Map<Integer, Integer> record = new HashMap<>();
for(int j=0;j<points.length;j++) {
int key = dis(points[i][0],points[i][1],points[j][0],points[j][1]);
if(j != i) {
if(record.containsKey(key)) {
int old_value = record.get(key);
int new_value = old_value + 1;
record.put(key, new_value);
}else {
record.put(key, 1);
}
}
}
//遍历Map中的value值
for(Map.Entry<Integer, Integer> entry : record.entrySet()) {
if(entry.getValue() >= 2) {
res += (entry.getValue())*(entry.getValue()-1);
}
}
}
return res;
}
//小陷阱——注意整型越界问题
//计算距离,为了避免溢出,不对其开根号
private static int dis(int a1, int a2, int b1, int b2) {
return (a1-b1)*(a1-b1) + (a2-b2)*(a2-b2);//返回距离的平方
}
}