力扣：3128. 直角三角形（Java）

题目描述：

给你一个二维 boolean 矩阵 grid 。
请你返回使用 grid 中的 3 个元素可以构建的 直角三角形 数目，且满足 3 个元素值 都 为 1 。
注意：
如果 grid 中 3 个元素满足：一个元素与另一个元素在 同一行，同时与第三个元素在 同一列 ，那么这 3 个元素称为一个 直角三角形 。这 3 个元素互相之间不需要相邻。

示例 ：

输入：grid = [[0,1,0],[0,1,1],[0,1,0]]

输出：2

解释：

有 2 个直角三角形。

代码实现：

class Solution {
    public long numberOfRightTriangles(int[][] grid) {
        int m = grid.length;// 二维矩阵的行数
        int n = grid[0].length;// 二维矩阵的列数
        int[] cols = new int[n];// 计算每一列的和
        for (int j = 0; j < n; j++) {// 先遍历列
            for (int i = 0; i < m; i++) {// 再遍历行
                cols[j] += grid[i][j];
            }
        }
        long res = 0;// 最终结果
        for (int i = 0; i < m; i++) {
            int rows = Arrays.stream(grid[i]).sum();// 计算每一行之和
            for (int j = 0; j < n; j++) {
                if (grid[i][j] == 1) {
                    // 如果当前元素存在两条直角边
                    res += (rows - 1) * (cols[j] - 1);// 将除元素本身的两组行，列元素集合进行组合，然后累加
                }
            }
        }
        return res;
    }
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23