1252. 奇数值单元格的数目
给你一个 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)
额外空间的算法来解决此问题吗?
方法一:计数
解题思路
模拟思路简单,但是复杂度上去了,就不写了。
计数:我们只需要用两个数组记录每一行和每一列「加一」的次数即可。然后统计奇数的坐标点即可。
参考代码 一
public int oddCells(int m, int n, int[][] indices) {
int[] rows = new int[m]; // 每一行 “加一” 的次数
int[] cols = new int[n]; // 每一列 “加一” 的次数
for (int[] index : indices) {
rows[index[0]]++;
cols[index[1]]++;
}
int count = 0;
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
if ((rows[i] + cols[j]) % 2 == 1) {
count++;
}
}
}
return count;
}
参考代码 二
官方的最终解法
- 计数的思路一致,同样是先统计每一行和每一列「加一」的次数
- 坐标点
[x, y]
为奇数的条件是:rows[x] 和 cols[y] 有且只有一个奇数; - 那最终的结果为:(
rows
中奇数的个数 *cols
中偶数的个数) + (cols
中奇数的个数 *rows
中偶数的个数)
public int oddCells2(int m, int n, int[][] indices) {
int[] rows = new int[m]; // 每一行 “加一” 的次数
int[] cols = new int[n]; // 每一列 “加一” 的次数
for (int[] index : indices) {
rows[index[0]]++;
cols[index[1]]++;
}
int oddX = 0;
for (int row : rows) {
if (row % 2 == 1) {
oddX++;
}
}
int oddY = 0;
for (int col : cols) {
if (col % 2 == 1) {
oddY++;
}
}
return oddX * (n - oddY) + oddY * (m - oddX);
}