统计有序矩阵中的负数
描述
给你一个 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
代码
最简单的就是暴力破解,但是这个因为有序,所以当我们找到是负数的时候,直接后面可以不判断了
class Solution(object):def countNegatives(self, grid):count = 0for i in range(len(grid)):for j in range(len(grid[0])):if grid[i][j] < 0:count += len(grid[0]) - jbreakreturn count
当然因为是不递增的,我们还可以想到使用二分查找法,时间复杂度是 O(logn)
class Solution(object):def getIndex(self, line):lineLen = len(line)left = 0right = lineLen - 1while left <= right:mid = (left + right) // 2if line[mid] < 0 and ((mid != 0 and line[mid -1] >= 0) or (mid == 0)):return lineLen - midelif line[mid] < 0:right = mid -1else:left = mid + 1return 0def countNegatives(self, grid):count = 0for i in range(len(grid)): count += self.getIndex(grid[i])return count