目录

lc447.回旋镖的数量

Shen 收录于每日一题

2021-09-13 约 152 字预计阅读 1 分钟

目录

447. 回旋镖的数量

计算两两点之间的距离
假如相同距离的点为$n$个的话，则满足条件的三元组有$A_{n}^{2} = n * (n - 1)$个

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21


class Solution {
public:
    // A_{n}^{2} = n * (n - 1)
    int numberOfBoomerangs(vector<vector<int>>& points) {
        int ans = 0;

        for(auto& p1: points){
            unordered_map<int, int> mp;
            for(auto& p2: points){
                int dis = (p1[0] - p2[0]) * (p1[0] - p2[0]) + (p1[1] - p2[1]) * (p1[1] - p2[1]);
                mp[dis] += 1;
            }
            for(auto& [_, cnt]: mp){
                cnt *= cnt - 1;
                ans += cnt;
            }
        }

        return ans;
    }
};