给你一个 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 {
public int oddCells(int n, int m, int[][] indices) {
int[][] arr = new int[n][m];
for(int i=0; i<indices.length; i++){
for(int j=0; j<m; j++) arr[indices[i][0]][j]++;
for(int j=0; j<n; j++) arr[j][indices[i][1]]++;
}
int ans = 0;
for(int i=0; i<n; i++){
for(int j=0; j<m; j++){
if(arr[i][j] % 2 == 1) ans++;
}
}
return ans;
}
}
显然某一行或某一列加了偶数次就忽略掉,奇数次的就是1
次,用odd_rows
和odd_cols
记录奇数次的行数和列数,相当于有odd_rows
行加了1
,odd_cols
列加了1
,奇数行和奇数列的交点出数字是2
,不交的是1
,其他的点都是0
,自行脑补,懒得画图。
也就是化简后有 o d d _ r o w s × o d d _ c o l s odd\_rows\times odd\_cols odd_rows×odd_cols 个点是2
,有 o d d _ r o w s × ( m − o d d _ c o l s ) + o d d _ c o l s × ( n − o d d _ r o w s ) odd\_rows \times (m - odd\_cols) + odd\_cols \times (n - odd\_rows) odd_rows×(m−odd_cols)+odd_cols×(n−odd_rows) 个点是1
,直接返回即可。
时间复杂度: O ( m a x ( n , m , l e n g t h ( i n d i c e s ) ) ) O(max(n,m,length(indices))) O(max(n,m,length(indices)))
python
运行的真慢!
class Solution:
def oddCells(self, n: int, m: int, indices: List[List[int]]) -> int:
rows = [0] * n
cols = [0] * m
for x, y in indices:
rows[x] += 1
cols[y] += 1
odd_rows = sum(row % 2 == 1 for row in rows)
odd_cols = sum(col % 2 == 1 for col in cols)
return odd_rows * (m - odd_cols) + odd_cols * (n - odd_rows)