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

素颜马尾好姑娘i 2023-01-10 03:52 121阅读 0赞

给你一个 m \* n 的矩阵 grid，矩阵中的元素无论是按行还是按列，都以非递增顺序排列。

请你统计并返回 grid 中 负数 的数目。

m == grid.length  
n == grid\[i\].length  
1 <= m, n <= 100  
\-100 <= grid\[i\]\[j\] <= 100

遍历每一行，二分找到当前遍历行的第一个负数出现的位置，即可求出当前行负数数量，之后遍历下一行时，第一个负数出现的位置一定在上一行负数出现位置或该位置之前，之后二分在该位置及之前找第一个负数即可得出改行中负数数量：

class Solution { 
    public:
        int findFirstNegative(vector<int> &ivec, int end) { 
            int start = 0;
            int mid = end >> 1;
            while (start <= end) { 
                if (ivec[mid] >= 0) { 
                    start = mid + 1;
                } else if (ivec[mid] < 0) { 
                    if (0 == mid) { 
                        return mid;
                    } else if (ivec[mid - 1] >= 0) { 
                        return mid;
                    } else { 
                        end = mid - 1;
                    }
                }
    
                mid = (start + end) >> 1;
            }
    
            return -1;
        }
    
        int countNegatives(vector<vector<int>>& grid) { 
            int ret = 0;
            int end = grid[0].size() - 1;
            for (vector<int> &ivec : grid) { 
                int index = findFirstNegative(ivec, end);
                if (index >= 0) { 
                    ret += ivec.size() - index;
                    end = index;
                }
            }
            return ret;
        }
    };

以上方法时间复杂度为O（mlogn），使用时可以将m和n中较大的当作n提高效率。