2482. Difference Between Ones and Zeros in Row and Column
Read the full problem statement on LeetCode.
Difficulty: medium Acceptance: 84% Topics: Array, Matrix, Simulation
View full problem on LeetCode Reading material
Reference solution (spoiler · python)
# Time: O(m * n)
# Space: O(m + n)
# array
class Solution(object):
def onesMinusZeros(self, grid):
"""
:type grid: List[List[int]]
:rtype: List[List[int]]
"""
rows = [sum(grid[i][j] for j in xrange(len(grid[0]))) for i in xrange(len(grid))]
cols = [sum(grid[i][j] for i in xrange(len(grid))) for j in xrange(len(grid[0]))]
return [[rows[i]+cols[j]-(len(grid)-rows[i])-(len(grid[0])-cols[j]) for j in xrange(len(grid[0]))] for i in xrange(len(grid))]
Solution from kamyu104/LeetCode-Solutions · MIT