如何将sorted list按右下角对角线填充转换为等块square matrix？

你的现有代码仅实现了列表的等长分割，完全没有处理从右下角开始沿对角线填充的核心逻辑，因此无法得到目标矩阵。

根据提示的栈思路，我们可以通过以下步骤实现需求：

• 先校验列表长度是否为n²（保证能构成n×n的方阵）
• 将列表元素存入栈，利用栈“后进先出”的特性，弹出元素时等价于从原列表末尾取数
• 遍历矩阵的所有对角线（从左上角的1元素对角线开始，依次到右下角的1元素对角线结束），将栈中弹出的元素依次填充到对应位置

实现代码 #

def create_matrix(lst, n):
    if len(lst) != n * n:
        raise ValueError("列表长度必须为n的平方，才能构成n×n方阵")
    # 将列表元素存入栈
    stack = lst.copy()
    # 初始化n×n的空矩阵
    matrix = [[0] * n for _ in range(n)]
    
    # 遍历所有对角线，共2n-1条
    for k in range(2 * n - 1):
        # 确定当前对角线的行索引范围
        if k < n:
            start_i, end_i = 0, k
        else:
            start_i, end_i = k - n + 1, n - 1
        
        # 填充当前对角线的每个位置
        for i in range(start_i, end_i + 1):
            j = k - i  # 计算列索引
            matrix[i][j] = stack.pop()
    
    return matrix

# 测试示例
mylist = [-2, -2, 1, 1, 4, 4, 3, 3, 3]
res = create_matrix(mylist, 3)
for row in res:
    print(row)

代码说明 #

• 栈的作用：原列表按元素出现次数升序排列，我们需要把出现次数多的元素（列表末尾的3）优先填充到矩阵左上角区域，用栈弹出刚好能实现反向取数
• 对角线遍历：通过参数k遍历所有2n-1条对角线，k从0到2n-2。对于每条对角线，根据k的大小确定行索引范围，再通过j = k - i计算对应列索引，确保填充位置在同一条对角线上
• 边界校验：提前判断列表长度是否符合方阵要求，避免出现无法分割的情况

输出结果 #

运行代码后会得到期望的矩阵：

[3, 3, 4]
[3, 4, 1]
[1, -2, -2]

内容的提问来源于stack exchange，提问作者meallhour