给你一个 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
示例 3:
输入:grid = [[1,-1],[-1,-1]]
输出:3
示例 4:
输入:grid = [[-1]]
输出:1
提示:
m == grid.length
n == grid[i].length
1 <= m, n <= 100
-100 <= grid[i][j] <= 100
解题思路
首先不难想到暴力,直接遍历即可。
class Solution:
def countNegatives(self, grid: List[List[int]]) -> int:
return sum(a < 0 for r in grid for a in r)
因为行和列以非递增顺序排列,所以我们可以从最后一行开始找,然后依次向上(上一层寻找的起始位置一定是大于等于当前层负数的起始位置)。
class Solution:
def countNegatives(self, grid: List[List[int]]) -> int:
r, c, res = len(grid), len(grid[0]), 0
m, n = r - 1, 0
while m >= 0 and n < c:
if grid[m][n] < 0:
res += c - n
m -= 1
else:
n += 1
return res
我将该问题的其他语言版本添加到了我的GitHub Leetcode
如有问题,希望大家指出!!!