文章目录
- 题目
- 思路
- 代码
题目
447. 回旋镖的数量
思路
问题要求计算平面上所有回旋镖的数量,即找到满足题设条件的点组合 (i, j, k)。回旋镖的定义是指有两个相同的距离,分别从点 i 到 j 和点 i 到 k。具体思路是:
遍历每个点,以其为中心,计算它与其他点的距离,并将这些距离存储下来。
对于每个点 i,遍历其余所有点 j,计算点 i 到点 j 的距离,并将距离作为键,出现的频次作为值,存储在哈希表中。
针对每个点 i,遍历哈希表中记录的距离和频次,如果某个距离有至少两个点与点 i 的距离相等,则可以形成回旋镖。对于每个满足条件的距离 d,假设有 m 个点与点 i 的距离等于 d,则回旋镖的数量为 m * (m - 1),因为对于第一个点有 m 种选择,对于第二个点有 m - 1 种选择。
统计所有点的回旋镖数量,并将其累加,最后得到所有回旋镖的总数。
代码
class Solution {
public:int numberOfBoomerangs(vector<vector<int>>& points) {int n = points.size(), res = 0;vector<unordered_map<int,int>>mapp(n);for(int i = 0;i < n; ++i){for(int j = i + 1; j < n; ++j){auto t = pow(points[i][0] - points[j][0], 2) + pow(points[i][1] - points[j][1], 2);res -= mapp[i][t] * (mapp[i][t] - 1);res -= mapp[j][t] * (mapp[j][t] - 1);mapp[i][t]++;mapp[j][t]++;res += mapp[i][t] * (mapp[i][t] - 1);res += mapp[j][t] * (mapp[j][t] - 1);}}return res;}
};