tyy 上课开小差，在本子上画了一个 n 行 n 列的矩阵。他把第一行 ~ 第 i 行 与 第 (n - i + 1) 行 ~ 第 n 行的所有元素交换位置。 他就得这样玩还不过瘾，想在对角上这么做，可是他能力有限请你帮助他完成这个任务。 你将获得一个 n 行 n 列的矩阵，并给你一个整数 k，当 k = 0 时为从左上到右下的主对角线，k = 1 时该对角线向右上平移一个单位长度，即向右上平移 k 个单位长度。 需要你将左上部分与右下部分对应的元素交换位置

好的，我来帮你解决这个问题。对于一个 n 行 n 列的矩阵，我们可以通过将第 i 行和第 (n - i + 1) 行的元素进行交换，将左上部分与右下部分对应的元素进行交换，实现对角线的翻转。具体操作如下：

当 k = 0 时，我们可以遍历矩阵的左下部分（即第二行到第 n 行，第一列到第 n - 1 列），并将其与右上部分（即第一行到第 n - 1 行，第二列到第 n 列）对应的元素进行交换。

当 k = 1 时，我们需要先将矩阵逆时针旋转 90 度，然后按照 k = 0 的方法进行操作，最后再将矩阵顺时针旋转 90 度，即可得到对角线向右上平移一个单位长度的结果。

下面是 Python 代码实现：

def diagonal_swap(matrix, k):
    n = len(matrix)
    if k == 1:
        # 逆时针旋转 90 度
        matrix = list(zip(*matrix[::-1]))
    for i in range(1, n):
        for j in range(i):
            if k == 0:
                # 左上部分与右下部分对应的元素进行交换
                matrix[i][j], matrix[j][i] = matrix[j][i], matrix[i][j]
            else:
                # 左下部分与右上部分对应的元素进行交换
                matrix[n - i][n - j - 1], matrix[n - j - 1][n - i] = matrix[n - j - 1][n - i], matrix[n - i][n - j - 1]
    if k == 1:
        # 顺时针旋转 90 度
        matrix = list(zip(*matrix[::-1]))
    return matrix

你可以将矩阵和 k 作为参数传递给该函数，函数将返回左上部分与右下部分对应的元素交换后的矩阵。