Special Positions in a Binary Matrix.py

// Runtime: 136 ms (Top 99.34%) | Memory: 17.90 MB (Top 5.36%)

class Solution:
    def numSpecial(self, mat: List[List[int]]) -> int:
        def get_column_sum(col_idx):
            return sum(row[col_idx] for row in mat)

        special = 0
        for row in mat:
            if sum(row) == 1:
                col_idx = row.index(1)
                special += get_column_sum(col_idx) == 1

        return special