力扣:3128. 直角三角形(Java)

avatar
作者
筋斗云
阅读量:0

目录

题目描述:

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

示例 :

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 个直角三角形。

代码实现:

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;     } } 

广告一刻

为您即时展示最新活动产品广告消息,让您随时掌握产品活动新动态!