记录每日LeetCode 2500.删除每行中的最大值 Java实现

小白的救赎2023-07-28 20:08

题目描述：

给你一个 m x n 大小的矩阵 grid ，由若干正整数组成。

执行下述操作，直到 grid 变为空矩阵：

从每一行删除值最大的元素。如果存在多个这样的值，删除其中任何一个。

将删除元素中的最大值与答案相加。

注意每执行一次操作，矩阵中列的数据就会减 1 。

返回执行上述操作后的答案。

初始代码：

java 复制代码

class Solution {
    public int deleteGreatestValue(int[][] grid) {

    }
}

示例1：

输入：grid = $\[1,2,4$ , $3,3,1$ ]

输出：8

解释：上图展示在每一步中需要移除的值。

在第一步操作中，从第一行删除 4 ，从第二行删除 3（注意，有两个单元格中的值为 3 ，我们可以删除任一）。在答案上加 4 。

在第二步操作中，从第一行删除 2 ，从第二行删除 3 。在答案上加 3 。

在第三步操作中，从第一行删除 1 ，从第二行删除 1 。在答案上加 1 。

最终，答案 = 4 + 3 + 1 = 8 。

示例2：

输入：grid = $\[10$ ]

输出：10

解释：上图展示在每一步中需要移除的值。

在第一步操作中，从第一行删除 10 。在答案上加 10 。

最终，答案 = 10 。

参考答案：

java 复制代码

// 暴力枚举 时间复杂度太高 
// 假设grid数组的一维长度是N,二维长度是M,对数组排序是logN
// 时间复杂度为O(N * N * M * logN)
class Solution {
    public int deleteGreatestValue(int[][] grid) {
        int sum = 0;
        for(int i = 0;i < grid[0].length; ++i) {
            sum += maxGrid(grid);
        }
        return sum;
    }

    public int maxGrid(int[][] grid) {
        int[] arr = new int[grid.length];
        // 记录最大值的索引
        int index = 0;
        int max = 0;
        for(int i = 0; i < grid.length; ++i) {
            max = grid[i][0];
            index = 0;
            for(int j = 0;j < grid[i].length; ++j) {
                // 遍历找到最大值 并且记录其索引
                max = Math.max(max, grid[i][j]);
                if(grid[i][j] == max) {
                    arr[i] = max;
                    index = j;
                }
                // 当列索引走到最后一个时 将最大值索引处的值置为0
                if(j == grid[i].length - 1) {
                    grid[i][index] = 0;
                }
            }
        }
        Arrays.sort(arr);
        return arr[grid.length - 1];
    }
}