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 个负数。
暴力肯定就是O(m * n)

对每一行进行二分查找,每次找出最左边第一个小于0的数,则该行之后的数都小于它,因为非增序排列。 时间复杂度O(m * log(n))
class Solution {
public:
    int countNegatives(vector>& grid) {
        int m = grid.size(), n = grid[0].size();
        int res = 0;
        for (int i = 0; i < m; ++i){
            int l = 0, r = n - 1;
            while(l < r){
                int mid = l + r >> 1;
                if (grid[i][mid] >= 0){
                    l = mid + 1;
                }else{
                    r = mid;
                }
            }
            if (grid[i][l] < 0)
                res += (n - l);
        }
        return res;
    }
};

 

你可能感兴趣的:(Leetcode)