【Leetcode】1351. Count Negative Numbers in a Sorted Matrix

题目地址:

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(1)O(1)

注解:
也可以用二分法来做,对每一行做二分,时间复杂度是O(nlogm)O(n\log m)

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章