LeetCode题目笔记——1351. 统计有序矩阵中的负数

题目描述

给你一个 m * n 的矩阵 grid，矩阵中的元素无论是按行还是按列，都以非递增顺序排列。 请你统计并返回 grid 中 负数 的数目。

 

示例 1：

输入：grid = [[4,3,2,-1],[3,2,1,-1],[1,1,-1,-2],[-1,-1,-2,-3]]
输出：8
解释：矩阵中共有 8 个负数。

示例 2：

输入：grid = [[3,2],[1,0]]
输出：0

 

提示：

• m == grid.length
• n == grid[i].length
• 1 <= m, n <= 100
• -100 <= grid[i][j] <= 100

 

进阶：你可以设计一个时间复杂度为 O(n + m) 的解决方案吗？

题目链接

题目难度——简单

方法一：暴力

  观察数据规模，可以发现最多也只有10000个元素，所以我们可以直接暴力解决。遍历整个数组，对负数进行计数。

代码/Python

class Solution:
    def countNegatives(self, grid: List[List[int]]) -> int:
        return sum(1 for row in grid for x in row if x < 0)

方法二：二分搜索

  注意题目疯狂暗示了矩阵是有序的，按行按列都是有序，所以我们可以对每一行用二分搜索，找到第一个非负数的位置，然后将答案加上n减去这个位置，这样的时间复杂度能做到O(MlogN)。

代码/Python

class Solution:
    def countNegatives(self, grid: list[list[int]]) -> int:
        res = 0
        m, n = len(grid), len(grid[0])
        for i in range(m):
            t = self.bina(grid[i], n)
            print(t)
            res += n - t if t != -1 else 0
        return res

    def bina(self, nums, n):
        left, right, pos = 0, n - 1, -1
        while left <= right:
            mid = left + (right - left) // 2
            if nums[mid] < 0:
                pos = mid
                right = mid - 1
            elif nums[mid] >= 0:
                left = mid + 1
        
        return pos

方法三：倒序遍历

  二分方法只用到了每行是有序的，我们还可以利用到按列也是有序的这一特点，也就是说，越往后面的行，负数出现的位置越靠前，所以我们倒序遍历，从最后一行开始，判断第一列是否是正数，是就往后遍历，直到遇到负数，再接着看上一行。

代码/Python

class Solution:
    def countNegatives(self, grid: list[list[int]]) -> int:
        m, n = len(grid), len(grid[0])
        res = 0
        j = 0
        for i in range(m - 1, -1, -1):
            while j < n and grid[i][j] >= 0:
                j += 1
            if j == n:
                break
            res += n - j

        return res

总结

  方法一时间复杂度O(N²)，空间复杂度O(1)， 方法二时间复杂度O(MlogN），空间O(1)，方法三时间O(M+N)，空间O（1）。