列表

详情


6367. 矩阵中的和

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

  1. 矩阵中每一行选取最大的一个数,并删除它。如果一行中有多个最大的数,选择任意一个并删除。
  2. 在步骤 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 。

 

提示:

原站题解

去查看

上次编辑到这里,代码来自缓存 点击恢复默认模板
class Solution { public: int matrixSum(vector<vector<int>>& nums) { } };

golang 解法, 执行用时: 116 ms, 内存消耗: 8.7 MB, 提交时间: 2023-05-14 17:14:25

func matrixSum(nums [][]int) (ans int) {
	for _, row := range nums {
		sort.Ints(row)
	}
	for j := range nums[0] {
		mx := 0
		for _, row := range nums {
			mx = max(mx, row[j])
		}
		ans += mx
	}
	return
}

func max(a, b int) int { if a < b { return b }; return a }

cpp 解法, 执行用时: 120 ms, 内存消耗: 46.8 MB, 提交时间: 2023-05-14 17:14:11

class Solution {
public:
    int matrixSum(vector<vector<int>> &nums) {
        for (auto &row: nums)
            sort(row.begin(), row.end());
        int ans = 0, n = nums[0].size();
        for (int j = 0; j < n; j++) {
            int mx = 0;
            for (auto &row: nums)
                mx = max(mx, row[j]);
            ans += mx;
        }
        return ans;
    }
};

java 解法, 执行用时: 14 ms, 内存消耗: 55.3 MB, 提交时间: 2023-05-14 17:13:58

class Solution {
    public int matrixSum(int[][] nums) {
        for (var row : nums)
            Arrays.sort(row);
        int ans = 0, n = nums[0].length;
        for (int j = 0; j < n; j++) {
            int mx = 0;
            for (var row : nums)
                mx = Math.max(mx, row[j]);
            ans += mx;
        }
        return ans;
    }
}

python3 解法, 执行用时: 100 ms, 内存消耗: 32.7 MB, 提交时间: 2023-05-14 17:13:45

class Solution:
    def matrixSum(self, nums: List[List[int]]) -> int:
        for row in nums: row.sort()
        return sum(map(max, zip(*nums)))  # zip(*nums) 枚举每一列

上一题