Leetcode 3128. Right Triangles

Espresso Macchiato2024-05-09 14:24

[Leetcode 3128. Right Triangles](#Leetcode 3128. Right Triangles)
- [1. 解题思路](#1. 解题思路)
- [2. 代码实现](#2. 代码实现)

题目链接：3128. Right Triangles

1. 解题思路

这一题的话对于任意一个位置，如果该位置为1，假设其所在行中1的个数 r i r_i ri，所在列中1的个数为 c j c_j cj，那么以该位置作为三角形顶点的三角形的个数就是 ( r i − 1 ) ⋅ ( c j − 1 ) (r_i-1)\cdot(c_j-1) (ri−1)⋅(cj−1)，因此，我们只需要先求出所有行和列当中的1的个数，然后遍历一下所有1的位置，将对应的三角形个数相加即可得到最终的答案。

2. 代码实现

给出python代码实现如下：

python 复制代码

class Solution:
    def numberOfRightTriangles(self, grid: List[List[int]]) -> int:
        n, m = len(grid), len(grid[0])
        rows = [sum(line) for line in grid]
        cols = [sum([grid[i][j] for i in range(n)]) for j in range(m)]
        ans = 0
        for i in range(n):
            for j in range(m):
                if grid[i][j] == 0:
                    continue
                ans += (rows[i]-1) * (cols[j]-1)
        return ans

提交代码评测得到：耗时2234ms，占用内存37.7MB。