Leetcode 2013. 检测正方形（Map）

给你一个在 X-Y 平面上的点构成的数据流。设计一个满足下述要求的算法：

添加 一个在数据流中的新点到某个数据结构中。可以添加 重复 的点，并会视作不同的点进行处理。
给你一个查询点，请你从数据结构中选出三个点，使这三个点和查询点一同构成一个 面积为正 的 轴对齐正方形 ，统计 满足该要求的方案数目。
轴对齐正方形 是一个正方形，除四条边长度相同外，还满足每条边都与 x-轴 或 y-轴 平行或垂直。

实现 DetectSquares 类：

DetectSquares() 使用空数据结构初始化对象
void add(int[] point) 向数据结构添加一个新的点 point = [x, y]
int count(int[] point) 统计按上述方式与点 point = [x, y] 共同构造 轴对齐正方形 的方案数。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/detect-squares
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

简单模拟，直接map存一下每个位置的点的数量，统计的时候枚举边长然后乘法原理计算并累加到答案即可。

class DetectSquares {
public:
    map<pair<int, int>, int> mp;
    DetectSquares() {

    }
    void add(vector<int> point) {
        pair<int, int> p = make_pair(point[0], point[1]);
        if(mp.find(p) != mp.end()) mp[p]++;
        else mp[p] = 1;
    }
    int count(vector<int> point) {
        int ans = 0;
        int nx = point[0], ny = point[1];
        for(int l = 1; l <= 1001; l++) {
            {
                pair<int, int> p1 = make_pair(nx, ny + l);
                pair<int, int> p2 = make_pair(nx + l, ny);
                pair<int, int> p3 = make_pair(nx + l, ny + l);
                if(mp.find(p1) != mp.end() && mp.find(p2) != mp.end() && mp.find(p3) != mp.end()) {
                    ans += mp[p1] * mp[p2] * mp[p3];
                }
            }
            {
                pair<int, int> p1 = make_pair(nx, ny - l);
                pair<int, int> p2 = make_pair(nx + l, ny - l);
                pair<int, int> p3 = make_pair(nx + l, ny);
                if(mp.find(p1) != mp.end() && mp.find(p2) != mp.end() && mp.find(p3) != mp.end()) {
                    ans += mp[p1] * mp[p2] * mp[p3];
                }
            }
            {
                pair<int, int> p1 = make_pair(nx - l, ny);
                pair<int, int> p2 = make_pair(nx - l, ny + l);
                pair<int, int> p3 = make_pair(nx, ny + l);
                if(mp.find(p1) != mp.end() && mp.find(p2) != mp.end() && mp.find(p3) != mp.end()) {
                    ans += mp[p1] * mp[p2] * mp[p3];
                }
            }
            {
                pair<int, int> p1 = make_pair(nx - l, ny - l);
                pair<int, int> p2 = make_pair(nx - l, ny);
                pair<int, int> p3 = make_pair(nx, ny - l);
                if(mp.find(p1) != mp.end() && mp.find(p2) != mp.end() && mp.find(p3) != mp.end()) {
                    ans += mp[p1] * mp[p2] * mp[p3];
                }
            }
        }
        return ans;
    }
};

/**
 * Your DetectSquares object will be instantiated and called as such:
 * DetectSquares* obj = new DetectSquares();
 * obj->add(point);
 * int param_2 = obj->count(point);
 */