LeetCode刷题笔记_598. 范围求和 II

题目描述

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

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

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

输入:
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。

思路

最大的那些数字一定出现在左上角，那么只要找到m和n的最小值进行相乘就好了。
需要一次遍历。

class Solution:
    def maxCount(self, m: int, n: int, ops: List[List[int]]) -> int:
        if len(ops) == 0:
            return m*n
        x = ops[0][0]
        y = ops[0][1]
        for op in ops:
            x = min(x, op[0])
            y = min(y, op[1])
        
        return x*y

题解

维护所有操作的交集

class Solution:
    def maxCount(self, m: int, n: int, ops: List[List[int]]) -> int:
        mina, minb = m, n
        for a, b in ops:
            mina = min(mina, a)
            minb = min(minb, b)
        return mina * minb

题解的方法与我的方法思路相同，但题解的代码书写更加优雅。

时间复杂度：O(k)，其中 k 是数组 ops 的长度。
空间复杂度：O(1)

参考资料

https://leetcode-cn.com/problems/range-addition-ii/