Leetcode 1351.统计有序矩阵中的负数（Count Negative Numbers in a Sorted Matrix）

Leetcode 1351.统计有序矩阵中的负数

1 题目描述（Leetcode题目链接）

  给你一个 m * n 的矩阵 grid，矩阵中的元素无论是按行还是按列，都以非递增顺序排列。

请你统计并返回 grid 中 负数 的数目。

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

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

提示：

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

2 题解

  对每一行进行二分查找。

class Solution:
    def countNegatives(self, grid: List[List[int]]) -> int:
        res = 0
        m, n = len(grid), len(grid[0])
        for i in range(m):
            l, r = 0, n
            while l < r:
                mid = (l+r)//2
                if grid[i][mid] >= 0:
                    l = mid + 1
                elif grid[i][mid] < 0:
                    r = mid
            res += n - l
        return res

二分查找调库

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

小于0的元素在矩阵中应该是一个阶梯形，因此从左下角开始往右上角遍历即可。

class Solution:
    def countNegatives(self, grid: List[List[int]]) -> int:
        res = 0
        m, n = len(grid)-1, len(grid[0])
        i = 0
        while m >= 0 and i < n:
            if grid[m][i] >= 0:
                i += 1
            else:
                res += n - i
                m -= 1
        return res