初学python记录:力扣1329. 将矩阵按对角线排序

题目:

矩阵对角线 是一条从矩阵最上面行或者最左侧列中的某个元素开始的对角线,沿右下方向一直到矩阵末尾的元素。例如,矩阵 mat63 列,从 mat[2][0] 开始的 矩阵对角线 将会经过 mat[2][0]mat[3][1]mat[4][2]

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

思考:

因为每一条对角线都彼此平行互不干扰,所以我们可以遍历每一条对角线上的元素,将其看成一个新数组,从小到大排序后填回原来矩阵中对应的位置。代码如下:

python 复制代码
class Solution(object):
    def diagonalSort(self, mat):
        """
        :type mat: List[List[int]]
        :rtype: List[List[int]]
        """
        m = len(mat)
        n = len(mat[0])

        def diagonal(x, y, mat):
            # 遍历对角线上所有元素,看成一个新数组,从小到大排序
            diagonal_number = []
            i = x
            j = y
            while i < m and j < n:
                diagonal_number.append(mat[i][j])
                i += 1
                j += 1
            diagonal_number.sort()
            
            # 将排序后的元素填回矩阵
            index = 0
            i_ = x
            j_ = y
            while i_ < m and j_ < n:
                mat[i_][j_] = diagonal_number[index]
                index += 1
                i_ += 1
                j_ += 1
            return mat

        diagonals = []
        # 将第一行和第一列所有坐标都计入对角线起点列表diagonals中
        for p in range(n):
            diagonals.append((0, p))
        for q in range(1, m):
            diagonals.append((q, 0))

        # 遍历每一条对角线
        for diag in diagonals:
            diagonal(diag[0], diag[1], mat)

        return mat

提交通过:

相关推荐
jyyyx的算法博客几秒前
Leetcode 2337 -- 双指针 | 脑筋急转弯
算法·leetcode
呵呵哒( ̄▽ ̄)"6 分钟前
线性代数:同解(1)
python·线性代数·机器学习
SweetCode12 分钟前
裴蜀定理:整数解的奥秘
数据结构·python·线性代数·算法·机器学习
程序员Linc24 分钟前
写给新人的深度学习扫盲贴:向量与矩阵
人工智能·深度学习·矩阵·向量
CryptoPP24 分钟前
springboot 对接马来西亚数据源API等多个国家的数据源
spring boot·后端·python·金融·区块链
ゞ 正在缓冲99%…25 分钟前
leetcode76.最小覆盖子串
java·算法·leetcode·字符串·双指针·滑动窗口
xcLeigh32 分钟前
OpenCV从零开始:30天掌握图像处理基础
图像处理·人工智能·python·opencv
大乔乔布斯32 分钟前
AttributeError: module ‘smtplib‘ has no attribute ‘SMTP_SSL‘ 解决方法
python·bash·ssl
惊鸿.Jh1 小时前
【滑动窗口】3254. 长度为 K 的子数组的能量值 I
数据结构·算法·leetcode
明灯L1 小时前
《函数基础与内存机制深度剖析:从 return 语句到各类经典编程题详解》
经验分享·python·算法·链表·经典例题