Leetcode 3128. Right Triangles

• Leetcode 3128. Right Triangles
  • 1. 解题思路
  • 2. 代码实现

• 题目链接：3128. Right Triangles

1. 解题思路

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

2. 代码实现

给出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。