447. 回旋镖的数量
1 <= n <= 500
由题目下的提示可以看出,本题的数据量并不大,所以本题可以使用“暴力”的解法,即遍历,并使用哈希表进行优化,
第一个for遍历,计算当前元素为首元素时,可能形成的排列组合;
第二个for遍历,计算当前元素与其他元素的距离,并以距离为索引存入该距离下的元素数量;
第三个for遍历,计算一定距离下的元素的排列组合数目。
class Solution {
public:
int numberOfBoomerangs(vector<vector<int>> &points) {
int ans = 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];
}
for (auto &m : cnt) {
ans += m.second * (m.second - 1);
}
}
return ans;
}
};