欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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

程序员文章站 2022-07-12 08:07:16
...

给你一个 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<vector<int>>& 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