【每日一题】2679. 矩阵中的和

2679. 矩阵中的和

题目描述

给你一个下标从 0 开始的二维整数数组 nums 。一开始你的分数为 0 。你需要执行以下操作直到矩阵变为空：

矩阵中每一行选取最大的一个数，并删除它。如果一行中有多个最大的数，选择任意一个并删除。
在步骤 1 删除的所有数字中找到最大的一个数字，将它添加到你的 分数 中。
请你返回最后的 分数 。

示例 1：

输入：nums = [[7,2,1],[6,4,2],[6,5,3],[3,2,1]]
输出：15
解释：第一步操作中，我们删除 7 ，6 ，6 和 3 ，将分数增加 7 。下一步操作中，删除 2 ，4 ，5 和 2 ，将分数增加 5 。最后删除 1 ，2 ，3 和 1 ，将分数增加 3 。所以总得分为 7 + 5 + 3 = 15 。

示例 2：

输入：nums = [[1]]
输出：1
解释：我们删除 1 并将分数增加 1 ，所以返回 1 。

提示：

1 <= nums.length <= 300
1 <= nums[i].length <= 500
0 <= nums[i][j] <= 103

解题思路

思路：堆。就直观模拟而言，我们需要先取每行最大数，然后将每行最大数进行比较求最大数，这样的话会存在很多中间变量，那么就需要一种数据结构来存储这些数据。考虑到每次取最大值，那么就可以利用大根堆，即一行使用一个大根堆，然后每行同步进行堆处理。

class Solution {
public:
    int matrixSum(vector<vector<int>>& nums) {
        int m=nums.size();
        int n=nums[0].size();
        //每一行一个大根堆
        vector<priority_queue<int>> pq(m);
        //每行一个大根堆
        for(int i=0;i<m;i++)
        {
            //每行存储列个
            for(int j=0;j<n;j++)
            {
                pq[i].emplace(nums[i][j]);
            }
        }
        int res=0;
        //总共列个循环
        for(int j=0;j<n;j++)
        {
            int maxlen=INT_MIN;
            //每行一个数
            for(int i=0;i<m;i++)
            {
                maxlen=max(maxlen,pq[i].top());
                pq[i].pop();
            }
            res+=maxlen;
        }
        return res;
    }
};

思路：排序。每行取最大数，然后按照列取行最大数中的最大数，其实就等价于先将行排序，再按照列依次取最大数。

class Solution {
public:
    int matrixSum(vector<vector<int>>& nums) {
        int m=nums.size();
        int n=nums[0].size();
        //每行排序
        for(int i=0;i<m;i++)
        {
            sort(nums[i].begin(),nums[i].end());
        }
        int res=0;
        //每列取一个最大数
        for(int j=0;j<n;j++)
        {
            int maxlen=INT_MIN;
            for(int i=0;i<m;i++)
            {
                maxlen=max(maxlen,nums[i][j]);
            }
            res+=maxlen;
        }
        return res;
    }
};

总结：一旦过程中存在很多中间变量，那么就需要考虑是否需要额外的符合数据特性和需求特性的数据结构来存储，而不是纯纯的比较。有时候模拟会导致数据处理起来很麻烦，那么可以考虑是否可以进行预处理，即不是根据题目要求的直接模拟思维，但是和题目分析等价的模拟，从而使得数据处理变得简单。