题目地址:
https://leetcode.com/problems/count-negative-numbers-in-a-sorted-matrix/
给定一个二维矩阵,每一行从左到右(非严格)单调下降,每一列从上到下(非严格)单调下降,求矩阵里有多少个负数。
从矩阵最右上角开始,先找到第一行的最后一个负数,然后累加第一行的负数个数,然后向下移动一格,由于每一列单调下降,所以一定会走到一个负数,然后向左走走到第二行最后一个负数,累加第二行负数个数。如此这样,一直走到最后一行为止。代码如下:
public class Solution {
public int countNegatives(int[][] grid) {
// 判空
if (grid == null || grid.length == 0 || grid[0].length == 0) {
return 0;
}
int res = 0;
int n = grid.length, m = grid[0].length;
// 从右上角开始出发
int x = 0, y = m - 1;
while (x < n) {
// 找第x行的最后一个负数
while (y >= 0 && grid[x][y] < 0) {
y--;
}
// 本行没有负数,则让y = m - 1,否则让y指向从左到右第一个负数
if (y < m - 1) {
y++;
}
// 如果本行有负数,那么累加本行负数的个数;
if (grid[x][y] < 0) {
res += m - y;
}
x++;
}
return res;
}
}
时间复杂度 O ( n + m ) O(n+m) O(n+m),空间 O ( 1 ) O(1) O(1)。
注解:
也可以用二分法来做,对每一行做二分,时间复杂度是
O
(
n
log
m
)
O(n\log m)
O(nlogm)。