PAT (Basic Level) Practice (中文)
1097 矩阵行平移
分数 20
作者 陈越
单位 浙江大学
给定一个 n×n 的整数矩阵。对任一给定的正整数 k<n,我们将矩阵的奇数行的元素整体向右依次平移 1、……、k、1、……、k、…… 个位置,平移空出的位置用整数 x 补。你需要计算出结果矩阵的每一列元素的和。
输入格式:
输入第一行给出 3 个正整数:n(<100)、k(<n)、x(<100),分别如题面所述。
接下来 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
输出样例:
529 481 479 263 417 342 343
样例解读
需要平移的是第 1、3、5、7 行。给定 k=2,应该将这三列顺次整体向右平移 1、2、1、2 位(如果有更多行,就应该按照 1、2、1、2、1、2 …… 这个规律顺次向右平移),左端的空位用 99 来填充。平移后的矩阵变成:
99 11 87 23 67 20 75
37 94 27 91 63 50 11
99 99 44 38 50 26 40
73 85 63 28 62 18 68
99 15 83 27 97 88 25
23 78 98 20 30 81 99
99 99 77 36 48 59 25
代码长度限制
16 KB
时间限制
400 ms
内存限制
64 MB
估计有更简单的方法,我看csdn没有,就上传了自己的O(∩_∩)O
注:记住是奇数行,但是i从0开始!
n,k,x=(int(i) for i in input().split())
t,r=[],[]
d=0
for i in range(n):
t.append(input().split())
r.append([0]*n)
for i in range(0,n):
if i%2==0:
m=d%k+1 #判断前进个数
for j in range(m):
r[i][j]=x
for j in range(n):
if j+m<n:
r[i][j+m]=t[i][j]
d+=1
else:
for j in range(n):
r[i][j]=t[i][j]
all=''
for i in range(n):
sum=0
for j in range(n):
d=int(r[j][i])
sum+=d
if i==n-1:
print(sum)
else:
print(sum,end=' ')