Leetcode每日一题 20240802 3128.直角三角形

题目描述

给你一个二维 boolean 矩阵 grid 。

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

注意:

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

3128.直角三角形

测试案例及提示

示例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的点,同时这三个点中需要满足一个元素与另一个元素在 同一行,同时与第三个元素在 同一列。
进一步分解问题

对于一个直角三角形,直角节点是唯一的,因此在寻找直角三角形的时候可以通过遍历直角节点,做到不重不漏。

对于一个直角节点来说,因为三角形的三个节点不需要相邻,因此在行中选取一个1,在列中再选取一个1,就可以组成直角三角形。
思路转换代码

先统计每行和每列中1的个数,使用两个数组存储。

遍历grid,对于每个值为1的,都统计以该节点为直角节点的直角三角形个数,不断加入答案中。
python

python 复制代码
class Solution:
    def numberOfRightTriangles(self, grid: List[List[int]]) -> int:
        ans = 0
        rows = len(grid)
        cols = len(grid[0])
        rowNum = [0] * rows
        colNum = [0] * cols
        for i in range(rows):
            for j in range(cols):
                if grid[i][j] == 1:
                    rowNum[i] += 1
                    colNum[j] += 1
        for i in range(rows):
            for j in range(cols):
                if grid[i][j] == 1:
                    ans += (rowNum[i] - 1) * (colNum[j] - 1)
        return ans
相关推荐
.似水4 分钟前
2025.4.22_C_可变参数列表
java·c语言·算法
天真小巫15 分钟前
2025.4.26总结
职场和发展
Felven24 分钟前
A. Ideal Generator
java·数据结构·算法
懒懒小徐32 分钟前
大厂面试-框架篇
面试·职场和发展
MoonBit月兔1 小时前
双周报Vol.70: 运算符重载语义变化、String API 改动、IDE Markdown 格式支持优化...多项更新升级!
ide·算法·哈希算法
How_doyou_do1 小时前
树状数组底层逻辑探讨 / 模版代码-P3374-P3368
数据结构·算法·树状数组
小鹿鹿啊1 小时前
C语言编程--14.电话号码的字母组合
c语言·开发语言·算法
小O的算法实验室2 小时前
2024年ESWA SCI1区TOP:量子计算蜣螂算法QHDBO,深度解析+性能实测
算法·论文复现·智能算法·智能算法改进
C语言魔术师2 小时前
509. 斐波那契数
算法·动态规划
共享家95272 小时前
栈相关算法题解题思路与代码实现分享
c++·leetcode