LeetCode——3128. 直角三角形

通过万岁!!!

  • 题目:给你一个二维数组,里面只有0或者1。然后让你找到里面的所有的由1构成的直角三角形。值得注意的是,三个1的位置直角的顶点与其他两点在同一行或者同一列。
  • 思路:这个题目其实就是确定好直角的那个1就好了,假设这一行有x个元素,这一列有y个元素,那么选定一个以后,还有x-1个,从这x-1个中随便选一个,然后从y-1个中随便选一个。那么这个点所构成的直角三角形的个数是(x-1)*(y-1)。
  • 技巧:组合

java代码

java 复制代码
class Solution {
    public long numberOfRightTriangles(int[][] grid) {
        Map<Integer, Integer> rowMap = new HashMap<>();// 第key行有多少个1
        Map<Integer, Integer> colMap = new HashMap<>();// 第key列有多少个1
        for (int i = 0; i < grid.length; i++) {
            for (int j = 0; j < grid[0].length; j++) {
                if (grid[i][j] == 1) {
                    Integer rowNum = rowMap.getOrDefault(i, 0);
                    rowMap.put(i, rowNum + 1);
                    Integer colNum = colMap.getOrDefault(j, 0);
                    colMap.put(j, colNum + 1);
                }
            }
        }
        long ret = 0;
        for (int i = 0; i < grid.length; i++) {
            if (rowMap.get(i) == null) {
                continue;
            }
            for (int j = 0; j < grid[0].length; j++) {
                if (grid[i][j] == 1) {
                    ret += (rowMap.get(i) - 1) * (colMap.get(j) - 1);
                }
            }
        }
        return ret;
    }
}

上面的代码时间和空间都不是很好。主要问题是这里没必要用map

java代码------进阶

java 复制代码
class Solution {
    public long numberOfRightTriangles(int[][] grid) {
        int n = grid.length;
        int m = grid[0].length;
        int[] colNum = new int[m];
        int[] rowNum = new int[n];
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < m; j++) {
                if (grid[i][j] == 1) {
                    colNum[j]++;
                    rowNum[i]++;
                }
            }
        }
        long ret = 0;
        for (int i = 0; i < n; i++) {
            if (rowNum[i] == 0) {
                continue;
            }
            for (int j = 0; j < m; j++) {
                if (grid[i][j] == 1) {
                    ret += (rowNum[i] - 1) * (colNum[j] - 1);
                }
            }
        }
        return ret;
    }
}
  • 总结:题目不是特别难,如果这个矩阵特别稀疏的话,我觉得还是map的方式会好一点。
相关推荐
淬渊阁3 分钟前
Go package
java·开发语言
CoderCodingNo13 分钟前
【GESP】C++二级真题 luogu-B4259 [GESP202503 二级] 等差矩阵
java·c++·矩阵
明月看潮生15 分钟前
青少年编程与数学 02-018 C++数据结构与算法 11课题、分治
c++·算法·青少年编程·编程与数学
佳腾_26 分钟前
【Web应用服务器_Tomcat】二、Tomcat 核心配置与集群搭建
java·前端·中间件·tomcat·web应用服务器
聂 可 以26 分钟前
IntelliJ IDEA修改实体类成员变量的名称(引入了该实体类的全部文件也会自动更新变量的名称)
java·ide·intellij-idea
冰茶_28 分钟前
C#中常见的设计模式
java·开发语言·microsoft·设计模式·微软·c#·命令模式
Echo``41 分钟前
2:QT联合HALCON编程—图像显示放大缩小
开发语言·c++·图像处理·qt·算法
.似水1 小时前
2025.4.22_C_可变参数列表
java·c语言·算法
禅与Bug的修复艺术1 小时前
JAVA后端开发常用的LINUX命令总结
java·linux·面试·java面试·后端开发·java后端·面试经验
佩奇的技术笔记1 小时前
Java学习手册:Java开发常用的内置工具类包
java