leetcode day 5 【 1351. 统计有序矩阵中的负数】从暴力到优化

相似题目:JZ4 二维数组中的查找

解题思路

暴力什么的就不搞了，直接上优化的算法。

• 思路：

从右上角开始，从右往左扫描，如果<0则其下面所有元素都<0（统计进来），直到碰到>0的，则该行查找完毕，跳入下一行。由于列和行都有规律，那么下一行不必从最右边开始，而是承接上一行的停止位来往前搜索，对每行依次执行直到达到退出条件（start=-1或所有行搜索完毕）。

• 复杂度分析：

对行来说，最多访问m次，对列来说，最多访问n列。所以复杂度应该是o(m+n)，可以看到比 暴力 o(mn) 或者 二分 o(mlogn) 提高得不要太多

• 代码如下

class Solution:
    def countNegatives(self, grid: List[List[int]]) -> int:
        count = 0
        start = len(grid[0]) - 1
        for i in range(len(grid)):
            while grid[i][start] < 0 and start >= 0:
                start = start - 1
                count = count + len(grid) - i
        return count

最终AC