598. 范围求和(Python)

题目

难度:★★☆☆☆
类型:数组,几何,益智

给定一个初始元素全部为 0,大小为 m*n 的矩阵 M 以及在 M 上的一系列更新操作。

操作用二维数组表示,其中的每个操作用一个含有两个正整数 a 和 b 的数组表示,含义是将所有符合 0 <= i < a 以及 0 <= j < b 的元素 M[i][j] 的值都增加 1。

在执行给定的一系列操作后,你需要返回矩阵中含有最大整数的元素个数。

注意
m 和 n 的范围是 [1,40000]。
a 的范围是 [1,m],b 的范围是 [1,n]。
操作数目不超过 10000。

示例

输入:
m = 3, n = 3
operations = [[2,2],[3,3]]
输出: 4

解释:
初始状态, M =
[[0, 0, 0],
[0, 0, 0],
[0, 0, 0]]

执行完操作 [2,2] 后, M =
[[1, 1, 0],
[1, 1, 0],
[0, 0, 0]]

执行完操作 [3,3] 后, M =
[[2, 2, 1],
[2, 2, 1],
[1, 1, 1]]

M 中最大的整数是 2, 而且 M 中有4个值为2的元素。因此返回 4。

解答

理论上,我们可以去模拟整个流程:

class Solution:
    def maxCount(self, m, n, ops):

        # 建立一个m行n列的零值矩阵
        create_matrix = lambda m, n: [[0 for _ in range(n)] for _ in range(m)]

        # 将指定区域的数字+1
        def add_one(matrix, r, c):              
            for i in range(r):
                for j in range(c):
                    matrix[i][j] += 1

        # 将矩阵展开成向量
        view = lambda matrix: [matrix[i][j] for i in range(len(matrix)) for j in range(len(matrix[0]))]

        # 获取向量中最大值出现的次数
        count_max = lambda vector: vector.count(max(vector))

        matrix = create_matrix(m, n)
        for r, c in ops:
            add_one(matrix, r, c)
            # print(matrix)

        return count_max(view(matrix))

但实际上,上述方法会超时,这个问题实际上很有规律,每次画出来一个框,就把框里所有元素都+1,而框的左上角位置一定是整个矩阵的左上角位置,某个位置有越多的框在此重合,这个位置最后的值越高,因此我们只需要把所有框的交集部分的元素个数求出来,即为所求。

这里需要另外考虑输入ops为空的情况,直接返回输入矩阵的面积就好。

class Solution:
    def maxCount(self, m, n, ops):
        return min([op[0] for op in ops]) * min([op[1] for op in ops]) if ops else m * n

如有疑问或建议,欢迎评论区留言~

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 在C语言中,五种基本数据类型存储空间长度的排列顺序是: A)char B)char=int<=float C)ch...
    夏天再来阅读 3,421评论 0 2
  • 基础篇NumPy的主要对象是同种元素的多维数组。这是一个所有的元素都是一种类型、通过一个正整数元组索引的元素表格(...
    oyan99阅读 5,164评论 0 18
  • 计算机二级C语言上机题库(南开版) 1.m个人的成绩存放在score数组中,请编写函数fun,它的功能是:将低于平...
    MrSunbeam阅读 6,483评论 1 42
  • NumPy是Python中关于科学计算的一个类库,在这里简单介绍一下。 来源:https://docs.scipy...
    灰太狼_black阅读 1,253评论 0 5
  • 这段时间社群的沸腾热度终于降下来了,即便是微信内测微信群拉人需群主确认的消息,也没有引起多大的反响。 浮躁始终是目...
    梵曦FANXI阅读 1,131评论 4 4