欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

LeetCode-Algorithms-[Easy][双百解法]1351. 统计有序矩阵中的负数

程序员文章站 2022-07-12 08:07:10
...

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

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

示例 1:

输入:grid = [[4,3,2,-1],[3,2,1,-1],[1,1,-1,-2],[-1,-1,-2,-3]]
输出:8
解释:矩阵*有 8 个负数。

示例 2:

输入:grid = [[3,2],[1,0]]
输出:0

示例 3:

输入:grid = [[1,-1],[-1,-1]]
输出:3

示例 4:

输入:grid = [[-1]]
输出:1

提示:

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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/count-negative-numbers-in-a-sorted-matrix
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。


最优解:倒序遍历

利用行列都非递增的性质,可以按行或按列遍历,官方题解给的复杂度我并不算完全赞同,我认为这个的复杂度应该为O(m+n)

    public int countNegatives(int[][] grid) {
        int  n = grid[0].length;
        int j = n - 1;
        int sum = 0;
        for (int[] column : grid) {
            while (j >= 0 && column[j] < 0) {
                j--;
            }
            sum += n - j - 1;
        }
        return sum;
    }

在此基础上的优化:二分查找

    public int countNegatives_BinarySearch(int[][] grid) {
        int n = grid[0].length;
        int sum = 0;
        int l = 0;
        int h = n - 1;
        for (int[] column : grid) {
            while (l < h) {
                int mid = l + ((h - l) >> 1);
                if (column[mid] >= 0) {
                    l = mid + 1;
                } else {
                    h = mid;
                }
            }
            if (column[l] < 0) {
                sum += n - l;
            }
            h = l;
            l = 0;
        }
        return sum;
    }

这个方法还能不能优化,实际上是可以的。因为m与n是有可能一个比另一个长的,我们保证拿比较大的那个数来进行二分,可以继续降低这个算法的时间复杂度

    public int countNegatives_BinarySearch_2(int[][] grid) {
        int m = grid.length;
        int n = grid[0].length;
        if (m > n) {
            return getSumWithLongerM(grid);
        }
        return getSumWithLongerN(grid);
    }

    private int getSumWithLongerM(int[][] grid) {
        int m = grid[0].length;
        int sum = 0;
        int l = 0;
        int h = m - 1;
        for (int[] row : grid) {
            while (l < h) {
                int mid = l + ((h - l) >> 1);
                if (row[mid] >= 0) {
                    l = mid + 1;
                } else {
                    h = mid;
                }
            }
            if (row[l] < 0) {
                sum += m - l;
            }
            h = l;
            l = 0;
        }
        return sum;
    }

    private int getSumWithLongerN(int[][] grid) {
        int n = grid[0].length;
        int sum = 0;
        int l = 0;
        int h = n - 1;
        for (int[] column : grid) {
            while (l < h) {
                int mid = l + ((h - l) >> 1);
                if (column[mid] >= 0) {
                    l = mid + 1;
                } else {
                    h = mid;
                }
            }
            if (column[l] < 0) {
                sum += n - l;
            }
            h = l;
            l = 0;
        }
        return sum;
    }