Leetcode-2500.[删除每行中的最大值]

34 阅读2分钟

题目描述: 给你一个 m x n 大小的矩阵 grid ,由若干正整数组成。

执行下述操作,直到 grid 变为空矩阵:

  • 从每一行删除值最大的元素。如果存在多个这样的值,删除其中任何一个。
  • 将删除元素中的最大值与答案相加。

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

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

示例 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

题解:

先对每行升序排序,这样每行的最大值就是行尾元素;每轮只需取所有行的行尾元素的最大值,累加后删除行尾即可,避免反复找最大值。时间复杂度:O (m*n logn)(排序每行的时间)

class Solution {
public:
    int deleteGreatestValue(vector<vector<int>>& grid) {
        if (grid.empty() || grid[0].empty()) return 0;
        int total = 0;
        int m = grid.size();
        int n = grid[0].size();

        // 第一步:对每行升序排序,最大值在每行末尾
        for (auto& row : grid) {
            sort(row.begin(), row.end());
        }

        // 第二步:每轮取所有行的最后一个元素(当前最大值),累加全局最大值
        for (int col = n - 1; col >= 0; --col) { // 从最后一列(最大值列)开始
            int round_max = 0;
            for (int row = 0; row < m; ++row) {
                round_max = max(round_max, grid[row][col]);
            }
            total += round_max;
        }

        return total;
    }
};