Java学习手册：（数据结构与算法-数组）Number of Boomerangs（leetcode447）

题目：

给出一个平面上的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);//返回距离的平方
	}
}