leetcode 1252. 奇数值单元格的数目（C++、python）

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

C++

class Solution {
public:
    int oddCells(int n, int m, vector>& indices) 
    {
        map s1;
        map s2;
        for(auto it:indices)
        {
            s1[it[0]]++;
            s2[it[1]]++;
        }
        int r=0;
        int c=0;
        for(auto x:s1)
        {
            if(x.second%2==1)
            {
                r++;
            }
        }
        for(auto x:s2)
        {
            if(x.second%2==1)
            {
                c++;
            }
        }
        int res=r*m+c*n-2*r*c;
        return res;
    }
};

python

class Solution:
    def oddCells(self, n: int, m: int, indices: List[List[int]]) -> int:
        dic1={}
        dic2={}
        for it in indices:
            if it[0] not in dic1:
                dic1[it[0]]=1
            else:
                dic1[it[0]]+=1
            if it[1] not in dic2:
                dic2[it[1]]=1
            else:
                dic2[it[1]]+=1
        r=0
        c=0
        for key in dic1:
            if 1==dic1[key]%2:
                r+=1
        for key in dic2:
            if 1==dic2[key]%2:
                c+=1
        return r*m+c*n-2*r*c