LeetCode题解之奇数值单元格的数目

奇数值单元格的数目

题目描述

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

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

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

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

示例 1:

示例图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:

示例图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

解题思路

首先，使用两个数组来分别记录行和列中各自的奇数个数，然后将行列中奇数个数相加然后减去重叠的部分即可得到答案。

复杂度分析

• 时间复杂度：O(m + n)。
• 空间复杂度：O(m + n)。

代码实现

class Solution {
    public int oddCells(int n, int m, int[][] indices) {
        boolean[] row = new boolean[n];
        boolean[] col = new boolean[m];

        for(int i = 0; i < indices.length; i ++){
            row[indices[i][0]] = !row[indices[i][0]];
            col[indices[i][1]] = !col[indices[i][1]];
        }

        int rowCount = 0, colCount = 0;
        for(int i = 0; i < row.length; i ++) {
            if(row[i]) {
                rowCount ++;
            }
        }

        for(int i = 0; i < col.length; i ++){
            if(col[i]) {
                colCount ++;
            }
        }
        
        return rowCount * m + colCount * n - rowCount * colCount * 2;
    }
}