Leetcode 问题：对角矩阵排序 |帮助但不要太多

如何解决Leetcode 问题：对角矩阵排序 |帮助但不要太多

问题：

矩阵对角线是一条单元格的对角线，从最顶行或最左列的某个单元格开始，沿右下角方向延伸，直到到达矩阵的末端。例如，从 mat[2][0] 开始的矩阵对角线，其中 mat 是一个 6 x 3 矩阵，包括单元格 mat[2][0]、mat3 和 mat[4][2] .

给定一个 m x n 整数矩阵 mat，按升序对每个矩阵对角线进行排序并返回结果矩阵。

我的问题：

我发现，对于一条特定的对角线，对角线单元格的行和列的差异是相同的。例如第 3,2,1 行的差异 (mat[0][0]... mat[i][i]) 显然是 0，除此之外它的 -1,-2 等等...

我可以提取差异

lst = [[3,3,1,1],[2,2],[1,2]]
new_list = []

for i in range(len(lst)):
    for j in range(len(lst[0])):
        if i-j < 0:
           new_list.append(lst[i][j])

print(new_list)
output -> [3,2]

到目前为止一切顺利，我想不出优先考虑特定差异的代码。我首先想要所有 -1 个单元格，然后是 -2 个单元格……等等。我可以写很多 if 语句，但这不是最佳解决方案…… 我几乎被困在这里。请帮助我提出一些新的想法。请不要告诉我完整的解决方案。把我推向正确的方向，感谢帮助！

解决方法

您可以做的是用对角线 ID 标识每个位置。然后按对角线 ID/值对展平的值进行排序，并将它们放回结果矩阵中：

def sortDiag(M):
    rows,cols = len(M),len(M[0])        # matrix dimensions
    diags     = range(rows + cols - 1)  # diagonal identifiers
    diagNo    = [ (d,r,c) for r in range(rows)
                          for c,d in enumerate(diags[-cols-r::][:cols]) ]
    diagVal   = [ (d,M[r][c]) for d,c in diagNo ] # values per diagonal
    result    = [[None]*cols for _ in range(rows)]
    for (_,v),(_,c) in zip(sorted(diagVal),sorted(diagNo)):
        result[r][c] = v  # assign position with sorted value
    return result
    

输出：

M = [[3,3,1,1],[2,2,2],[1,2]]
print(*M,sep="\n")

[3,1]
[2,2]
[1,2]

print(*sortDiag(M),sep="\n")

[1,1]
[1,3]

在此示例中，对角线标识符是从 0 到 5 的数字，diagNo 中的坐标列表对应于以下分配：

[2,4,5]
[1,4]
[0,3]

通过将排序位置（每对角线）与排序值（每对角线）进行映射，我们可以将排序值分配给每个对角线对应的位置。