题目:
给定平面上 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
思路:(暴力枚举+哈希表,排列问题)
固定一个点,求出剩余点到该点的距离,满足条件的个数为:(因为要考虑顺序是个排列问题)
所以遍历数组,重复上述步骤,将<距离,个数>暂存在一个unordered_map<int, int>(底层是哈希表实现,查找更高效些),最后遍历哈希表,累加满足条件的个数
代码:
class Solution {
public:
//用一个哈希表存储两点距离
int numberOfBoomerangs(vector<vector<int>>& points) {
//记录回旋镖个数
int num = 0;
for (auto& p:points) {
//存<距离,个数>,因为顺序问题,回旋标个数牵扯到排列问题,
unordered_map<int, int> res;
for (auto &q :points) {
//题目已说是互不相同的点,等于0只有一个(本身到本身)
int dis = (p[0] - q[0]) * (p[0] - q[0]) + (p[1] - q[1]) * (p[1] - q[1]);
++res[dis];
}
for (auto& k : res) {
num += k.second * (k.second - 1);
}
}
return num;
}
};