力扣每日一题 直角三角形

题目

中等 数学

给你一个二维 boolean 矩阵 grid

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

注意:

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

示例 1:

|---|---|---|
| 0 | 1 | 0 |
| 0 | 1 | 1 |
| 0 | 1 | 0 |

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

**输出:**2

解释:

有 2 个直角三角形。

示例 2:

|---|---|---|---|
| 1 | 0 | 0 | 0 |
| 0 | 1 | 0 | 1 |
| 1 | 0 | 0 | 0 |

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

**输出:**0

解释:

没有直角三角形。

示例 3:

|---|---|---|
| 1 | 0 | 1 |
| 1 | 0 | 0 |
| 1 | 0 | 0 |

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

**输出:**2

解释:

有两个直角三角形。

思路

根据题意,我们遍历这个二维数组,遇到为1的就计算它这行和这一列还有多少个1然后个数相乘就是可以组合出来的直角的数量,刚开始我的思路有点问题,我想的是计算当前遍历到的1的右边有多少1和下边有多少1,它们的数量相乘,可是它的上边和左边的1也是可以和后面的1组合出直角的,所以应该直接用当前行和当前列的1的数量-1再相乘即可。

解题过程

为了简化计算,我们可以预先设置两个数组计算出每一行和每一列的1的个数,然后挨个遍历数组,如果当前为1则结果加上(当前行的1的数量-1)*(当前列的1的数量-1),这样遍历一遍即可

复杂度

  • 时间复杂度: O(n^2)
  • 空间复杂度: O(n)

代码如下:

cpp 复制代码
class Solution {
public:
    
    long long numberOfRightTriangles(vector<vector<int>>& grid) {
        int n=grid.size(),m=grid[0].size();
        vector<int> col(m),row(n);//分别存储每一行和每一列的1的个数
        for(int i=0;i<grid.size();i++)
        {
            for(int j=0;j<grid[i].size();j++)
            {
                col[j]+=grid[i][j];
                row[i]+=grid[i][j];
            }
        }
        long long res=0;

        for(int i=0;i<n;i++)
        {
            for(int j=0;j<m;j++)
            {
                if(grid[i][j]==1)
                {
                    res+=(row[i]-1)*(col[j]-1);
                }
            }
        }

        return res;
    }
};
相关推荐
知来者逆20 分钟前
计算机视觉——速度与精度的完美结合的实时目标检测算法RF-DETR详解
图像处理·人工智能·深度学习·算法·目标检测·计算机视觉·rf-detr
阿让啊25 分钟前
C语言中操作字节的某一位
c语言·开发语言·数据结构·单片机·算法
এ᭄画画的北北25 分钟前
力扣-160.相交链表
算法·leetcode·链表
爱研究的小陈1 小时前
Day 3:数学基础回顾——线性代数与概率论在AI中的核心作用
算法
渭雨轻尘_学习计算机ing1 小时前
二叉树的最大宽度计算
算法·面试
BB_CC_DD2 小时前
四. 以Annoy算法建树的方式聚类清洗图像数据集,一次建树,无限次聚类搜索,提升聚类搜索效率。(附完整代码)
深度学习·算法·聚类
梁下轻语的秋缘3 小时前
每日c/c++题 备战蓝桥杯 ([洛谷 P1226] 快速幂求模题解)
c++·算法·蓝桥杯
CODE_RabbitV3 小时前
【深度强化学习 DRL 快速实践】逆向强化学习算法 (IRL)
算法
mit6.8244 小时前
[贪心_7] 最优除法 | 跳跃游戏 II | 加油站
数据结构·算法·leetcode
keep intensify4 小时前
通讯录完善版本(详细讲解+源码)
c语言·开发语言·数据结构·算法