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

给你一个 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提高效率。