解题思路:
可以先统计每一行和每一列的 1 的个数,记录在数组 rows 和 cols 中。
然后我们枚举每一个 1,假设当前 1 在第 i 行第 j 列,那么以当前 1 为直角三角形的直角点,另外两个直角点分别在第 i 行和第 j 列,那么直角三角形的个数就是 (rows[i]−1)×(cols[j]−1),累加到答案中即可。
java
class Solution {
public long numberOfRightTriangles(int[][] grid) {
//如果
int r =grid.length;
int c =grid[0].length;
int[] rows = new int[r];
int[] columns = new int[c];
for(int i=0;i<r;i++){
for(int j=0;j<c;j++){
if(grid[i][j]==1){
rows[i]++;
columns[j]++;
}
}
}
long count = 0;//注意如果不使用long型,在遇到大的数字时正数会变成负数。
for(int i=0;i<r;i++){
for(int j=0;j<c;j++){
if(grid[i][j]==1&&(rows[i]>=1)&&(columns[j]>=1)){
count+=(long)(rows[i]-1)*(columns[j]-1);
}
}
}
return count;
}
}