【LeetCode:3128. 直角三角形 + 枚举 + 乘法原理】-CSDN博客

本文链接：https://blog.csdn.net/Coder_ljw/article/details/140863461

在这里插入图片描述

🚀 算法题 🚀

🌲 算法刷题专栏 | 面试必备算法 | 面试高频算法 🍀
🌲 越难的东西,越要努力坚持，因为它具有很高的价值，算法就是这样✨
🌲 作者简介：硕风和炜，CSDN-Java领域优质创作者🏆，保研|国家奖学金|高中学习JAVA|大学完善JAVA开发技术栈|面试刷题|面经八股文|经验分享|好用的网站工具分享💎💎💎
🌲 恭喜你发现一枚宝藏博主,赶快收入囊中吧🌻
🌲 人生如棋，我愿为卒，行动虽慢，可谁曾见我后退一步？🎯🎯

🚀 算法题 🚀

在这里插入图片描述

🍔 目录

🚩 题目链接

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

🌟 求解思路&实现代码&运行结果

⚡ 枚举 + 乘法原理

🥦 求解思路

枚举直角三角形的中间的点，找到每一行每一列1的个数，根据乘法原理求得直角三角形中1的个数，返回即可。
提示：为什么想到乘法原理？一个元素与另一个元素在同一行，同时与第三个元素在同一列，那么这 3 个元素称为一个直角三角形
有了基本的思路，接下来我们就来通过代码来实现一下的解法。

🥦 实现代码

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