矩阵中的和 2679 Medium

Problem: 2679. 矩阵中的和

思路

假设$m=len(nums),n=len(nums[0])$，每次要寻找$m$行第$k(k<=n)$大的数，所以就直接对每一行进行排序，之后寻找更新。

解题方法

如果$m==1$，那么返回值就为第$0$行的元素之和。
如果$m>1$，就需要对比每一行的第$k(k<=n>)$大的数字，我们首先把第一行的元素进行排序，第$j$列就表示第$k$大的数字，之后就遍历所有行，对每一列的数字进行更新，找到每一列对应的最大值。返回该数组的总和即可。

复杂度

• 时间复杂度: $O(n^{2}logn)$

• 空间复杂度: $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)