上课开小差，在本子上画了一个 n 行 n 列的矩阵。他把第一行 ~ 第 i 行 与 第 (n - i + 1) 行 ~ 第 n 行的所有元素交换位置。 他就得这样玩还不过瘾，想在对角上这么做，可是他能力有限请你帮助他完成这个任务。 你将获得一个 n 行 n 列的矩阵，并给你一个整数 k，当 k = 0 时为从左上到右下的主对角线，k = 1 时该对角线向右上平移一个单位长度，即向右上平移 k 个单位长度。 需要你将左上部分与右下部分对应的元素交换位置。 输入格式 输入两个整数 n (1 <= n <= 100) 和 k （ 0 <= k < n ） 接下来 n 行 n 列输入矩阵（0 <= 矩阵元素 <= 100） 输出格式 输出处理后的矩阵。 输入样例 复制 4 1 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 输出样例 复制 1 5 9 13 2 6 10 14 3 7 11 15 4 8 12 16

这道题的思路有点绕，但是细节不多，只要理解了题目的意思，就能写出来。我们可以按照如下的步骤来实现：

1. 如果 k=0，则对于矩阵中第 i 行第 j 列的元素，如果 i<j，则将其与第 j 行第 i 列的元素交换位置。

2. 如果 k>0，则对于矩阵中第 i 行第 j 列的元素，如果 i+j<n+1-k，则将其与第 j+k 行第 i+k 列的元素交换位置。

下面是对应的 Python 代码：