Problem: 2679. 矩阵中的和
思路
假设 m = l e n ( n u m s ) , n = l e n ( n u m s [ 0 ] ) m=len(nums),n=len(nums[0]) m=len(nums),n=len(nums[0]),每次要寻找 m m m行第 k ( k < = n ) k(k <= n) k(k<=n)大的数,所以就直接对每一行进行排序,之后寻找更新。
解题方法
如果 m = = 1 m==1 m==1,那么返回值就为第 0 0 0行的元素之和。
如果 m > 1 m>1 m>1,就需要对比每一行的第 k ( k < = n > ) k(k<=n>) k(k<=n>)大的数字,我们首先把第一行的元素进行排序,第 j j j列就表示第 k k k大的数字,之后就遍历所有行,对每一列的数字进行更新,找到每一列对应的最大值。返回该数组的总和即可。
复杂度
-
时间复杂度: O ( n 2 l o g n ) O(n^2logn) O(n2logn)
-
空间复杂度: O ( n ) O(n) O(n)
Code
class Solution:
def matrixSum(self, nums: List[List[int]]) -> int:
m = len(nums)
n = len(nums[0])
# 如果只有一行,那么返回值就是sum(nums[0])
# 如果有两行,那么需要看每次操作的数字
# 如果第二行的第k(k < n)大的数字比第一行第k大的数字大,那么返回值需要更改
res = nums[0][:] # 首先设置返回值为第一行的总和
res.sort()
for i in range(1, m):
nums[i].sort() # 要对比顺序相对应的数字
for j in range(n):
if res[j] < nums[i][j]:
res[j] = nums[i][j]
return sum(res)