leetcode 周赛 162 5255. 奇数值单元格的数目

目录

1. 解答

给你一个 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

解答

分别统计各行各列加一的次数，然后遍历矩阵，行列次数加一就是每个位置的加一次数。

class Solution:
    def oddCells(self, n: int, m: int, indices: List[List[int]]) -> int:
        line = [0 for i in range(n)]
        col =  [0 for i in range(m)]
        for x in indices:
            line[x[0]] += 1
            col[x[1]] += 1
        cnt = 0
        for i in range(n):
            for j in range(m):
                if (line[i] + col[j]) % 2 == 1:
                    cnt += 1
        return cnt