给你一个下标从 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
思路:排序
因为每次是在每行中依次选取最大值,所以可以对每行进行排序
用max_col保存每列的最大值
class Solution:
def matrixSum(self, nums: List[List[int]]) -> int:
# 对每行进行排序,然后选取每列最大值累加到res中
n = len(nums)
# 记录每列的最大值
max_col = []
# m:有多少列
m = len(nums[0])
for i in range(n):
# 先对每一行进行排序,这里用到了sorted函数,即不改变原数组的值
row = sorted(nums[i])
if i == 0:
max_col = row
else:
for j in range(m):
if row[j] > max_col[j]:
max_col[j] = row[j]
return sum(max_col)