LeetCode 750. 角矩形的数量 （枚举、哈希表）

750. 角矩形的数量
关键是如何枚举矩形。
暴力枚举就是分别枚举矩形的两个角，时间复杂度：$O(n^4)$

下面的做法，依次考虑每增加的一行对答案的贡献。
具体来说，
如果上面已经在 3 行在列索引为1，9出现，
那么现在这一行也在列索引1，9出现，那么对答案贡献了 3 次。

class Solution {
public:
    int countCornerRectangles(vector<vector<int>>& grid) {
        int ans = 0;
        int n = grid[0].size();
        vector<int> mp(40010);
        for(auto &row:grid) {
            for(int i = 0; i < n; i++) {
                if(!row[i]) continue;
                for(int j = i + 1; j < n; j++) {
                    if(!row[j]) continue;
                    int pos = i*200 + j;
                    ans += mp[pos];
                    mp[pos]++;
                }
            }
        }
        return ans;
    }
};

普通的枚举方式。

class Solution {
public:
    int countCornerRectangles(vector<vector<int>>& grid) {
        int m = grid.size(), n = grid[0].size();
        int rs[210], cs[210];
        int c1, c2; 
        int ans = 0;
        for(int i = 0; i < m; i++) {
            for(int j = 0; j < n; j++) {
                if(!grid[i][j]) continue;
                c1 = c2 = 0;
                for(int x = i + 1; x < m; x++) {
                    if(grid[x][j]) {
                        rs[c1++] = x;
                    }
                }
                for(int y = j + 1; y < n; y++) {
                    if(grid[i][y]) {
                        cs[c2++] = y;
                    }
                }
                for(int k1 = 0; k1 < c1; k1++) {
                    for(int k2 = 0; k2 < c2; k2++) {
                        if(grid[rs[k1]][cs[k2]]) {
                            ans++;
                        }
                    }
                }
            }
        }
        return ans;
    }
};