开启掘金成长之旅！这是我参与「掘金日新计划 · 12 月更文挑战」的第31天，点击查看活动详情

题目描述

1252. 奇数值单元格的数目 - 力扣（LeetCode）

给你一个 m x n 的矩阵，最开始的时候，每个单元格中的值都是 0。

另有一个二维索引数组 indices，indices[i] = [ri, ci] 指向矩阵中的某个位置，其中 ri 和 ci 分别表示指定的行和列（从 0 开始编号）。

对 indices[i] 所指向的每个位置，应同时执行下述增量操作：

ri 行上的所有单元格，加 1 。 ci 列上的所有单元格，加 1 。 给你 m、n 和 indices 。请你在执行完所有 indices 指定的增量操作后，返回矩阵中 奇数值单元格 的数目。

示例 1：

输入：m = 2, n = 3, indices = [[0,1],[1,1]]
输出：6
解释：最开始的矩阵是 [[0,0,0],[0,0,0]]。
第一次增量操作后得到 [[1,2,1],[0,1,0]]。
最后的矩阵是 [[1,3,1],[1,3,1]]，里面有 6 个奇数。

示例 2：

输入：m = 2, n = 2, indices = [[1,1],[0,0]]
输出：0
解释：最后的矩阵是 [[2,2],[2,2]]，里面没有奇数。

 

提示：

• 1 <= m, n <= 50
• 1 <= indices.length <= 100
• 0 <= ri < m
• 0 <= ci < n  

进阶：你可以设计一个时间复杂度为 O(n + m + indices.length) 且仅用 O(n + m) 额外空间的算法来解决此问题吗？

思路分析

只有奇数次的操作才是有效操作

所以用row，col记录每行每列的操作，当操作次数为奇数记录为1，否则记录为0

记录完所有的操作后，只需要计算row，col中1的个数

r * m + c * n表示奇数的个数，不过还需要去掉行列的交点，交点个数为r * c

因为行列都计算了一次交点，所以需要乘2

AC 代码

class Solution:
    def oddCells(self, n: int, m: int, indices: List[List[int]]) -> int:
        # result = [[0] * m for _ in range(n)]
        row = [0] * n
        col = [0] * m
        for ri, ci in indices:
            row[ri] = 1 - row[ri]
            col[ci] = 1 - col[ci]
        r, c = row.count(1), col.count(1)
        return r * m + c * n - r * c * 2