个人技术分享

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) (ri1)(cj1),因此,我们只需要先求出所有行和列当中的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。