LeetCode_1351_统计有序矩阵中的负数

题目链接

• https://leetcode-cn.com/problems/count-negative-numbers-in-a-sorted-matrix/

解题思路

• 暴力法
• 二分查找法
• 通过二分查找可以找到每一行中从前往后的第一个负数，那么这个位置之后导这一行的末尾里所有的数必然是负数，可以直接统计
  • 遍历矩阵的每一行
  • 二分查找到改行从前往后的第一个负数，考虑第i行，我们记这个位置为$po{s}_i$，那么第i行[$po{s}_i,m-1$]中的所有数都是负数，所以这一行对答案的贡献就是m - 1 - $po{s}_i$ + 1 = m - $po{s}_i$
  • 最后的答案就是${\sum }_{i=0}^{n-1}(m-po{s}_i)$

AC代码

//暴力法
class Solution {
    public int countNegatives(int[][] grid) {
        int ans = 0;
        for (int[] value : grid) {
            for (int j = 0; j < grid[0].length; j++) {
                if (value[j] < 0)
                    ans++;
            }
        }
        return ans;
    }
}

//二分查找
class Solution {
public:
    int countNegatives(vector<vector<int>>& grid) {
        int num=0;
        for (auto x:grid){
            int l=0,r=(int)x.size()-1,pos=-1;
            while (l<=r){
                int mid=l+((r-l)>>1);
                if (x[mid]<0){
                    pos=mid;
                    r=mid-1;
                }
                else l=mid+1;
            }
            if (~pos) num+=(int)x.size()-pos;// pos=-1表示这一行全是>=0的数，不能统计
        }
        return num;
    }
};

本地测试代码

package com.company;

public class Solution_1351 {
    public static int countNegatives(int[][] grid) {
        int ans = 0;
        for (int[] value : grid) {
            for (int j = 0; j < grid[0].length; j++) {
                if (value[j] < 0)
                    ans++;
            }
        }
        return ans;
    }

    public static void main(String[] args) {
        System.out.println(countNegatives(new int[][]{{4, 3, 2, -1}, {3, 2, 1, -1}, {1, 1, -1, -2}, {-1, -1, -2, -3}}));
        System.out.println(countNegatives(new int[][]{{3, 2}, {1, 0}}));
        System.out.println(countNegatives(new int[][]{{1, -1}, {-1, -1}}));
        System.out.println(countNegatives(new int[][]{{-1}}));
    }
}