LeetCode 598 Java实现

LeetCode598

题目

给定一个初始元素全部为 0，大小为 m*n 的矩阵 M 以及在 M 上的一系列更新操作。

操作用二维数组表示，其中的每个操作用一个含有两个正整数 a 和 b 的数组表示，含义是将所有符合 0 <= i < a 以及 0 <= j < b 的元素 M[i][j] 的值都增加 1。

在执行给定的一系列操作后，你需要返回矩阵中含有最大整数的元素个数。

示例 1:

输入:
m = 3, n = 3
operations = [[2,2],[3,3]]
输出: 4
解释:
初始状态, M =
[[0, 0, 0],
[0, 0, 0],
[0, 0, 0]]

执行完操作 [2,2] 后, M =
[[1, 1, 0],
[1, 1, 0],
[0, 0, 0]]

执行完操作 [3,3] 后, M =
[[2, 2, 1],
[2, 2, 1],
[1, 1, 1]]

M 中最大的整数是 2, 而且 M 中有4个值为2的元素。因此返回 4。
注意:

m 和 n 的范围是 [1,40000]。
a 的范围是 [1,m]，b 的范围是 [1,n]。
操作数目不超过 10000。

代码

思路还是比较清晰的,注意审题就好
这道题其实我刚拿到手上的第一反应是照着这个做一遍…累加过去
但是想了想真的实在太慢了…
然后换个思路想,其实求最多的个数,就是求最小的i和j而已(因为大于他们的都会给他们加上,他们是每次都累加的

class Solution {
    public int maxCount(int m, int n, int[][] ops) {
        int i = 0;
        int j = 0;
        int[] a = new int[ops.length];
        int[] b = new int[ops.length];
        int c = 0;
        for (int w[]:ops
        //用两个数组来分别存储i和j!
             ) {
            a[c] = w[0];
            b[c] = w[1];
            c++;
        }
        Arrays.sort(a);
        Arrays.sort(b);
        //进行排序来获得最小的 i 和 j
        //如果没获得到就是默认的0啦
        for (int w: a
        ) {
            if( w != 0)
            {
                i = w;
                break;
            }
        }
        for (int w: b
             ) {
            if( w != 0)
            {
                j = w;
                break;
            }
        }
        //接下来就是判断了, 脑补一下! 如果 m <= i的话是不是就相当于只有m行会被加
        //同理n <= j也是!
        //如果 m > i的话,就只有i行被加啦
        if(i == 0 && j == 0)
        {
            return m*n;
        }
        if( m <= i && n <= j){
            return m*n;
        }
        if( m <= i && n > j){
            return m*j;
        }
        if(m > i && n > j){
            return i*j;
        }
        if( m > i && n <= j){
            return i*n;
        }

        throw new IndexOutOfBoundsException("no return");
    }
}