Number of Boomerangs

題目來源
給一堆點坐標,讓你求出一堆的三個點{a, b, c}集合,其中ab距離和ac距離一樣。
我只能想到暴力的方法,先把每一個點都哈希一下存起來,然后再遍歷,取兩點,判斷對應的第三點是否存在。
抄的代碼,代碼如下:

class Solution {
public:
    int numberOfBoomerangs(vector<pair<int, int>>& points) {
        int res = 0;
        int n = points.size();
        for (int i=0; i<n; i++) {
            unordered_map<double, int> map;
            for (int j=0; j<n; j++) {
                int deltaX = points[i].first - points[j].first;
                int deltaY = points[i].second - points[j].second;
                res += 2 * map[hypot(deltaX, deltaY)]++;
            }
        }
        return res;
    }
};

重點在于計算res的過程,2+4+6+8+...+2n = n(n+1)。
map先申請好,這樣子快了一些:

class Solution {
public:
    int numberOfBoomerangs(vector<pair<int, int>>& points) {
        int res = 0;
        int n = points.size();
        for (int i=0; i<n; i++) {
            unordered_map<double, int> map(n-1);
            for (int j=0; j<n; j++) {
                if (i == j)
                    continue;
                int deltaX = points[i].first - points[j].first;
                int deltaY = points[i].second - points[j].second;
                res += 2 * map[deltaX*deltaX+deltaY*deltaY]++;
            }
        }
        return res;
    }
};
最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。

推薦閱讀更多精彩內容