1. 程式人生 > >LeetCode 447. 迴旋鏢的數量

LeetCode 447. 迴旋鏢的數量

給定平面上 n 對不同的點,“迴旋鏢” 是由點表示的元組 (i, j, k) ,其中 i 和 j 之間的距離和 i 和 k 之間的距離相等(需要考慮元組的順序)。

找到所有迴旋鏢的數量。你可以假設 n 最大為 500,所有點的座標在閉區間 [-10000, 10000] 中。

示例:

輸入:
[[0,0],[1,0],[2,0]]

輸出:
2

解釋:
兩個迴旋鏢為 [[1,0],[0,0],[2,0]][[1,0],[2,0],[0,0]]

解題思路:

暴力算出所有的兩點之間的距離,

對於一個點a,有n個點到點a的距離相等,則對a來說有n*(n-1)個迴旋鏢(多看幾個就找出規律了)

程式碼: 

class Solution {
public:
    int numberOfBoomerangs(vector<pair<int, int>>& points) {
        int ans = 0;
        for(int i = 0;i < points.size(); i ++){
            map<int, int> m;
            for(int j = 0; j < points.size(); j ++){
                if(i == j) continue;
                int d = (points[i].first-points[j].first)*(points[i].first-points[j].first)+(points[i].second-points[j].second)*(points[i].second-points[j].second);
                m[d]++;
            }
            map<int, int>::iterator mi;
            for(mi = m.begin(); mi != m.end(); mi ++){
                ans += mi->second*(mi->second-1);
            }
        }
        return ans;
    }
};