对角线打印 4*4 的矩阵

Question

以下是一个对角线打印 4*4 矩阵的示例代码：

matrix = [
    [1, 2, 3, 4],
    [5, 6, 7, 8],
    [9, 10, 11, 12],
    [13, 14, 15, 16]
]

m = len(matrix)
n = len(matrix[0])

for k in range(m+n-1):
    if k < n:
        i = 0
        j = k
    else:
        i = k - n + 1
        j = n - 1

    while i < m and j >= 0:
        print(matrix[i][j])
        i += 1
        j -= 1

运行上述代码将输出以下内容：

1
5
2
9
6
3
13
10
7
4
14
11
8
15
12
16

这个代码中，我们使用了两个指针 i 和 j 来遍历矩阵的对角线。对于每个对角线，我们从矩阵的第一行开始，每次将 i 增加 1，j 减少 1，直到到达矩阵的最后一行或最后一列。这样，我们就可以按照对角线顺序打印矩阵中的元素。