问题描述
给定平面上 n 对 互不相同 的点 points ,其中 points[i] = [xi, yi] 。回旋镖 是由点 (i, j, k) 表示的元组 ,其中 i 和 j 之间的距离和 i 和 k 之间的距离相等(需要考虑元组的顺序)。
返回平面上所有回旋镖的数量。
输入输出
示例 1:
输入:points = [[0,0],[1,0],[2,0]]
输出:2
解释:两个回旋镖为 [[1,0],[0,0],[2,0]] 和 [[1,0],[2,0],[0,0]]
示例 2:
输入:points = [[1,1],[2,2],[3,3]]
输出:2
示例 3:
输入:points = [[1,1]]
输出:0
提示:
n == points.length
1 <= n <= 500
points[i].length == 2
-104 <= xi, yi <= 104
所有点都 互不相同
笔记
#include<unordered_map>
//哈希表初始化
unordered_map<int, int> cnt = {
{2, 3},
{3, 4},
{1, 4},
{5, 10}
};
//取哈希表的value(倒叙?)
for (auto &[_, m] : cnt) {
cout << m << endl;
}
代码
class Solution {
public:
int numberOfBoomerangs(vector<vector<int>>& points) {
int ans = 0;
//两个指针 [0, 0] : [1, 0] & [1, 0] : [0, 0] 是两个距离
for(auto &p : points){
// 距离 : 个数
unordered_map<int, int> cnt;
for(auto &q :points){
//存放距离(避免开方,同意存成平方)
int dis = (p[0] - q[0])*(p[0] - q[0]) + (p[1] - q[1])*(p[1] - q[1]);
++cnt[dis];
}
//倒叙遍历cnt,获得m????
for(auto &[_, m] : cnt){
//m 个元素中选出 2 个不同元素的排列数
//[0, 0] 本身到自己的距离也计算了,所以 0 : 1 - > 1*(1-1)
ans += m*(m-1);
}
}
return ans;
}
};