阅读量:0
解题思路:
可以先统计每一行和每一列的 1 的个数,记录在数组 rows 和 cols 中。
然后我们枚举每一个 1,假设当前 1 在第 i 行第 j 列,那么以当前 1 为直角三角形的直角点,另外两个直角点分别在第 i 行和第 j 列,那么直角三角形的个数就是 (rows[i]−1)×(cols[j]−1),累加到答案中即可。
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; } }