题目描述:
给你一个 m x n 大小的矩阵 grid ,由若干正整数组成。
执行下述操作,直到 grid 变为空矩阵:
从每一行删除值最大的元素。如果存在多个这样的值,删除其中任何一个。
将删除元素中的最大值与答案相加。
注意 每执行一次操作,矩阵中列的数据就会减 1 。
返回执行上述操作后的答案。
初始代码:
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 。
参考答案:
// 暴力枚举 时间复杂度太高
// 假设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];
}
}