- Leetcode 3128. Right Triangles
- 1. 解题思路
- 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代码实现如下:
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 = 0for i in range(n):for j in range(m):if grid[i][j] == 0:continueans += (rows[i]-1) * (cols[j]-1)return ans
提交代码评测得到:耗时2234ms,占用内存37.7MB。