LeetCode-Python-5152. 将矩阵按对角线排序(数组 + 排序)

给你一个 m * n 的整数矩阵 mat ,请你将同一条对角线上的元素(从左上到右下)按升序排序后,返回排好序的矩阵。

 

示例 1:

输入:mat = [[3,3,1,1],[2,2,1,2],[1,1,1,2]]
输出:[[1,1,1,1],[1,2,2,2],[1,2,3,3]]

 

提示:

  • m == mat.length
  • n == mat[i].length
  • 1 <= m, n <= 100
  • 1 <= mat[i][j] <= 100

思路:

按照题意,把所有对角线上的元素都提取到一个数组里,排好序,再放回矩阵中。

时间复杂度:O(MN + KlogK),K = max(M,N)

空间复杂度:  O(K)

class Solution(object):
    def diagonalSort(self, mat):
        """
        :type mat: List[List[int]]
        :rtype: List[List[int]]
        """
        if not mat or not mat[0]:
            return None

        m, n = len(mat), len(mat[0])

        for j in range(n):
            i, l, pos = 0, [], []
            while i < m and j < n:
                l.append(mat[i][j])
                pos.append([i, j])
                i += 1
                j += 1

            l.sort()
            for i, p in enumerate(pos):
                x, y = p
                mat[x][y] = l[i]
        
        for i in range(1, m):
            j, l, pos = 0, [], []
            while i < m and j < n:
                l.append(mat[i][j])
                pos.append([i, j])
                i += 1
                j += 1

            l.sort()
            for i, p in enumerate(pos):
                x, y = p
                mat[x][y] = l[I]

        return mat

 

发布了722 篇原创文章 · 获赞 106 · 访问量 19万+

猜你喜欢

转载自blog.csdn.net/qq_32424059/article/details/104085451