题目描述
给你一个 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]],里面没有奇数。
提示:
解法
由题目可知,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