持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第22天,点击查看活动详情
一、题目描述:
1252. 奇数值单元格的数目 - 力扣(LeetCode)
给你一个 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) 额外空间的算法来解决此问题吗?
二、思路分析:
首先行和列做的顺序不重要,我们就先做行的再做列的。
这一行总共做了几次也不重要,关键是这一行做了奇数次还是偶数次,偶数次这一行就全是偶数,奇数次这一行就全变奇数了。
然后再去考虑列,那些做了偶数次的列的对最后结果不影响,可以无视,奇数次的列对之前行处理完的结果有影响,所以我们只要统计有几个列是做了奇数次,例如有x列。
那么我们从行的角度来看,如果这一行是本全是偶数,最后就有x个奇数,如果本来全是奇数,那实际上就是有m-x个奇数。
三、AC 代码:
class Solution {
public:
int oddCells(int m, int n, vector<vector<int>>& indices) {
vector<int> rowOdd(m, 0), colOdd(n, 0);
for (auto &rc: indices) {
int r = rc[0], c = rc[1];
rowOdd[r] = 1 - rowOdd[r];
colOdd[c] = 1 - colOdd[c];
}
int oddCol = 0, ans = 0;
for (int i = 0; i < n; ++i) oddCol += colOdd[i];
for (int i = 0; i < m; ++i) ans += rowOdd[i] == 1 ? n - oddCol : oddCol;
return ans;
}
};