题目
给你一个下标从 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 。
解题
分析题目以及示例可知,这道题的实质就是选每行最大的值相加的到最后的分数,因此只需要对行排序后取出每列的最大值相加即可。
代码
class Solution(object):
def matrixSum(self, nums):
"""
:type nums: List[List[int]]
:rtype: int
"""
point=0
for i in nums:
i.sort()
nums = list(zip(*nums))
for i in nums:
point += max(i)
return point
解题的过程中使用到了zip()函数,可以看下面这个例子理解zip()函数:
a = [1,2,3]
b = [4,5,6]
c = [4,5,6,7,8]
zipped = zip(a,b) # 打包为元组的列表
[(1, 4), (2, 5), (3, 6)]
zip(a,c) # 元素个数与最短的列表一致
[(1, 4), (2, 5), (3, 6)]
zip(*zipped) # 与 zip 相反,*zipped 可理解为解压,返回二维矩阵式
[(1, 2, 3), (4, 5, 6)]
zip()可以理解为压缩 而zip(*)可以理解为解压的过程
大佬的文章让我受益匪浅:https://blog.csdn.net/PaulZhn/article/details/104391756