【013】Leetcode—数组—1252. 奇数值单元格的数目(Sum of Square Numbers)

题目信息

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

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

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

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

说明:

  • 1 <= n <= 50
  • 1 <= m <= 50
  • 1 <= indices.length <= 100
  • 0 <= indices[i][0] < n
  • 0 <= indices[i][1] < m

示例:

1:

【013】Leetcode—数组—1252. 奇数值单元格的数目(Sum of Square Numbers)_第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:

【013】Leetcode—数组—1252. 奇数值单元格的数目(Sum of Square Numbers)_第2张图片

输入:n = 2, m = 2, indices = [[1,1],[0,0]]
输出:0
解释:最后的矩阵是 [[2,2],[2,2]],里面没有奇数。

解题思路

        #改变矩阵的函数
        def changeMatrix(num):
            #改变行
            for i in range(m):
                matrix[num[0]][i] += 1
            #改变列
            for i in range(n):
                matrix[i][num[1]] += 1
        
        #计算list奇数个数
        def countOdd(num):
            return len([x for x in num if x%2 != 0])

        matrix = []
        for i in range(n):
            matrix += [[0]*m]

        for item in indices:
            changeMatrix(item)

        result = 0
        for item in matrix:
            result += countOdd(item)

        return result

主要思路是建立2个子函数,重点说一下第二个,用列表推导式直接快速求出奇数个数,然后通过遍历去累加结果

学习

TODO
思路比较简单,可以考虑先处理indices中重复的地方,然后再进行矩阵加 甚至可以没有实际意义上的矩阵,比如重复加两次第n行,那么实际上等于没加,可以推广到偶数等于0次,奇数等于1次,这样可以减少无意义的加法操作

你可能感兴趣的:(霍乱时期的Python之路)