题目:
给你一个二维 boolean 矩阵 grid
。
请你返回使用 grid
中的 3 个元素可以构建的 直角三角形 数目,且满足 3 个元素值 都 为 1 。
注意:
- 如果
grid
中 3 个元素满足:一个元素与另一个元素在 同一行,同时与第三个元素在 同一列 ,那么这 3 个元素称为一个 直角三角形 。这 3 个元素互相之间不需要相邻。
示例 1:
0 | 1 | 0 |
0 | 1 | 1 |
0 | 1 | 0 |
0 | 1 | 0 |
0 | 1 | 1 |
0 | 1 | 0 |
输入:grid = [[0,1,0],[0,1,1],[0,1,0]]
输出:2
解释:
有 2 个直角三角形。
提示:
1 <= grid.length <= 1000
1 <= grid[i].length <= 1000
0 <= grid[i][j] <= 1
思路:
寻找直角三角形的个数,关键就在一个点:如何去重
我们访问二维数组中的每一个点,如果为1,那么就可以成为直角顶点,这样在访问不同 矩阵位置的时候,就是以不同的直角顶点为基本,这样就不用考虑去重问题了
统计二维矩阵每一行,每一列的1的个数,这样当 grid[i][j] 为1时,以它为直角顶点的直角三角形个数就是 (该行 1 的个数 - 1) * (该列 1 的个数 - 1)
代码:
class Solution {
public:
long long numberOfRightTriangles(vector<vector<int>>& grid) {
int m = grid.size(), n = grid[0].size();
vector<int> rowCnt(m, 0);
vector<int> colCnt(n, 0);
for (int i = 0; i < m; i++) {
int cnt = 0;
for (int j = 0; j < n; j++) {
if (grid[i][j]) cnt++;
}
rowCnt[i] = cnt;
}
for (int j = 0; j < n; j++) {
int cnt = 0;
for (int i = 0; i < m; i++) {
if (grid[i][j]) cnt++;
}
colCnt[j] = cnt;
}
long long ans = 0;
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
if (grid[i][j]) {
ans += (rowCnt[i] - 1) * (colCnt[j] - 1);
}
}
}
return ans;
}
};