Java实现LeetCode447.回旋镖的数量

题目描述

给定平面上 n 对 互不相同 的点 points ，其中 points[i] = [xi, yi] 。回旋镖 是由点 (i, j, k) 表示的元组 ，其中 i 和 j 之间的距离和 i 和 k 之间的欧式距离相等（ 需要考虑元组的顺序 ）。

返回平面上所有回旋镖的数量。

思路

首先，看到数据量，判断应该可以暴力+模拟，似乎也没有什么别的好方法。

先开始想的是数据结构是Map<Integer,List<int[][]>>，其中key为点之间的距离distance，val结构是list<int[][]>，放具体两点的坐标。那么经过两层循环后，示例一在map中的数据如下：

显然，val里存在顺序不同的元组对，还需要对val进行循环判断，遂放弃。

看了题解以后，发现同样是双重循环，但是先将回旋镖的 中间点a 定下来，维护一个map，记录到该中间点距离为某定值的点 b 的数量，进行排列组合后累加。

代码

class Solution {
    public int numberOfBoomerangs(int[][] points) {
        int len = points.length;
        int ans = 0;
        for( int i = 0 ; i < len ; i++ ){
            Map<Integer, Integer> mp = new HashMap();
            for( int j = 0 ; j < len ; j++ ){
                if( i==j ) continue;
                int x = points[i][0] - points[j][0];
                int y = points[i][1] - points[j][1];
                int dis = x * x + y * y;
                mp.put( dis , mp.getOrDefault( dis , 0 ) + 1 );
            }
            for( int key : mp.keySet() ){
                int val = mp.get( key );
                ans += ( val*(val-1 ) );
            }
        }
        return ans;
    }
}

反思

学习使用map的getOrDefault函数，可以避免if else讨论map中是否存在key