PTA矩阵列平移 python

给定一个 n×n 的整数矩阵。对任一给定的正整数 k

输入格式：

输入第一行给出 3 个正整数：n（<100）、k（

接下来 n 行，每行给出 n 个不超过 100 的正整数，为矩阵元素的值。数字间以空格分隔。

输出格式：

在一行中输出平移后第 1 到 n 行元素的和。数字间以 1 个空格分隔，行首尾不得有多余空格。

输入样例：

7 2 99
11 87 23 67 20 75 89
37 94 27 91 63 50 11
44 38 50 26 40 26 24
73 85 63 28 62 18 68
15 83 27 97 88 25 43
23 78 98 20 30 81 99
77 36 48 59 25 34 22

输出样例：

440 399 369 421 302 386 428

样例解读

需要平移的是第 2、4、6 列。给定 k=2，应该将这三列顺次整体向下平移 1、2、1 位（如果有更多列，就应该按照 1、2、1、2 …… 这个规律顺次向下平移），顶端的空位用 99 来填充。平移后的矩阵变成：

11 99 23 99 20 99 89
37 87 27 99 63 75 11
44 94 50 67 40 50 24
73 38 63 91 62 26 68
15 85 27 26 88 18 43
23 83 98 28 30 25 99
77 78 48 97 25 81 22

python代码

def fz(m,k):
    z1 = []
    for j in range(k):
        z = []
        for i in m:
            z.append(i[j])
        z1.append(z)
    return z1

def cl(m,k,z,y):
    d = 0
    for i in range(z):
        if i%2==1:
            d+=1
            for e in range(d):
                del m[i][-1]
                m[i].insert(0,y)
        if d==k:
            d=0
    return m

m = list(map(int,input().split()))
k = []
for i in range(m[0]):
    k.append(list(map(int,input().split())))
k = fz(k,m[0])
k = cl(k,m[1],m[0],m[2])
k = fz(k,m[0])
a = []
for i in k:
#     print(i)
    a.append(str(sum(i)))
print(' '.join(a))