LeetCode：3128. 直角三角形（枚举 Java）

目录

3128. 直角三角形

题目描述：

实现代码与解析：

枚举

原理思路：

3128. 直角三角形

题目描述：

        给你一个二维 boolean 矩阵 grid 。

请你返回使用 grid 中的 3 个元素可以构建的 直角三角形 数目，且满足 3 个元素值 都 为 1 。

注意：

• 如果 grid 中 3 个元素满足：一个元素与另一个元素在 同一行，同时与第三个元素在 同一列 ，那么这 3 个元素称为一个 直角三角形 。这 3 个元素互相之间不需要相邻。

示例 1：

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

输出：2

解释：

有 2 个直角三角形。

示例 2：

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

输出：0

解释：

没有直角三角形。

示例 3：

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

输出：2

解释：

有两个直角三角形。

提示：

• 1 <= grid.length <= 1000
• 1 <= grid[i].length <= 1000
• 0 <= grid[i][j] <= 1

实现代码与解析：

枚举

class Solution {     public long numberOfRightTriangles(int[][] grid) {          int n = grid.length;         int m = grid[0].length;          int[] col = new int[m];          long res = 0L;         for (int i = 0; i < n; i++) {             for (int j = 0; j < m; j++) {                 col[j] += grid[i][j];             }         }         for (int i = 0; i < n; i++) {             int row = Arrays.stream(grid[i]).sum();             for (int j = 0; j < m; j++) {                 if (grid[i][j] == 1) {                     res += (col[j] - 1) * (row - 1);                 }             }          }         return res;     } }

原理思路：

        比较简单，枚举直角的交叉点，交叉点对应行和对应列的 1 的个数相乘求和就是答案。当然要减去交叉点，所以 - 1 再相乘。