Leetcode 1252. 奇数值单元格的数目

题目描述

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

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

你需要将每对 [ri, ci] 指定的行和列上的所有单元格的值加 1。

请你在执行完所有 indices 指定的增量操作后，返回矩阵中 「奇数值单元格」 的数目。

示例 1:

输入：n = 2, m = 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:

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

提示:

• 1 <= n <= 50
• 1 <= m <= 50
• 1 <= indices.length <= 100
• 0 <= indices[i][0] < n
• 0 <= indices[i][1] < m

解法

由题目可知，n 行 m 列的二维数组，每一行的增量操作会影响 m 个元素，每一列的增量操作会影响 n 个元素，因为最终要计算的是奇数的个数，而初始数值为偶数，所以不妨计算元素的增量操作次数即可，若为奇数次，则元素最终为奇数。

因为每次操作都是直接影响一行或者一列元素，所以不妨计算最终影响了多少行元素，与多少列元素。相乘再相加，即得出最终影响的个数，因为行列相遇相当于不操作，所以最后去除多加的元素数。

对同一行或同一列执行两次操作相当于不执行操作

class Solution:
    def oddCells(self, n: int, m: int, indices: List[List[int]]) -> int:
        rset,cset=set(),set()
        for r,c in indices:
            if r in rset:
                rset.remove(r)
            else:
                rset.add(r)
            if c in cset:
                cset.remove(c)
            else:
                cset.add(c)
        rnum,cnum=len(rset),len(cset)
        return rnum*m+cnum*n-2*rnum*cnum