1582. Special Positions in a Binary Matrix 链接到标题 直接遍历二维数组之后,如果某个位置是 1,再去统计每行的和和每列的和是否为 1 时间复杂度太高,可以先统计每行每列的和,然后再遍历二维数组。
class Solution: def numSpecial(self, mat: List[List[int]]) -> int: m, n, ans = len(mat), len(mat[0]), 0 row, col = [0] * m, [0] * n for i in range(m): for j in range(n): if mat[i][j]: row[i] += 1 col[j] += 1 pool = [j for j in range(n) if col[j] == 1] for i in range(m): if row[i] !