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

目录

题目描述:

给你一个二维 boolean 矩阵 grid 。

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

注意:

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

示例 :

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

输出:2

解释:

有 2 个直角三角形。

代码实现:

java 复制代码
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;
    }
}
相关推荐
秋说43 分钟前
【PTA数据结构 | C语言版】一元多项式求导
c语言·数据结构·算法
Maybyy1 小时前
力扣61.旋转链表
算法·leetcode·链表
一只叫煤球的猫1 小时前
【🤣离谱整活】我写了一篇程序员掉进 Java 异世界的短篇小说
java·后端·程序员
斐波娜娜1 小时前
Maven详解
java·开发语言·maven
Bug退退退1231 小时前
RabbitMQ 高级特性之事务
java·分布式·spring·rabbitmq
程序员秘密基地1 小时前
基于html,css,vue,vscode,idea,,java,springboot,mysql数据库,在线旅游,景点管理系统
java·spring boot·mysql·spring·web3
皮皮林5512 小时前
自从用了CheckStyle插件,代码写的越来越规范了....
java
小码氓2 小时前
Java填充Word模板
java·开发语言·spring·word
会飞的天明2 小时前
Java 导出word 实现饼状图导出--可编辑数据
java·word
Muxiyale2 小时前
使用spring发送邮件,部署ECS服务器
java·服务器·spring