Leetcode 1329. 将矩阵按对角线排序 题解

首先遍历对角线元素，顺序为：

1. 先从第一列的最后一行到第一行
2. 然后从第一行的第一列到最后一列

遍历的同时记录坐标和数值，对数值进行排序，然后坐标顺序放回。

class Solution:
    def diagonalSort(self, mat: List[List[int]]) -> List[List[int]]:
        # 从第一列的最后一行到第一行
        for j in range(len(mat[0])-1, -1, -1):
            cords = [] # 记录坐标顺序
            nums = [] # 记录数值
            i = 0
            while(i < len(mat) and j < len(mat[0])):
                cords.append([i, j])
                nums.append(mat[i][j])
                i += 1
                j += 1
            nums.sort() # 数值排序
            for k in range(len(nums)):
                cord_i, cord_j = cords[k]
                mat[cord_i][cord_j] = nums[k] # 按坐标顺序放回
        # 从第一行的第一列到最后一列
        for i in range(1, len(mat)):
            cords = []
            nums = []
            j = 0
            while(i < len(mat) and j < len(mat[0])):
                cords.append([i, j])
                nums.append(mat[i][j])
                i += 1
                j += 1
            nums.sort()
            for k in range(len(nums)):
                cord_i, cord_j = cords[k]
                mat[cord_i][cord_j] = nums[k]
        return mat