PTA乙级 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

| 代码长度限制 | 时间限制 | 内存限制 |
| 16KB | 400ms | 64MB |

思路：
①读入二维数组，建议下标从1开始，后面会方便判断奇数行
②定义flag等于1，遇到奇数行则进行平移，用flag判断该平移几位，当一整行全部平移结束后flag++，若flag大于k，则flag再变回1，[1,k]为flag值的一个循环
③当平移结束后用sum计算每一列的总值并按题目格式输出

代码：

#include<bits/stdtr1c++.h>
using namespace std;
int nums[105][105];
int main() {
	int n, k, x, i, j;
	cin >> n >> k >> x;
	for (i = 1; i <= n; i++) {
		for (j = 1; j <= n; j++) {
			scanf("%d", &nums[i][j]); //读入二维数组
		}
	}
	int flag = 1;
	for (i = 1; i <= n; i++) {
		if (i % 2 != 0) { //此时为奇数行，需要平移
			for (j = n; j >= flag + 1; j--) nums[i][j] = nums[i][j - flag];
			for (j = 1; j <= flag; j++) nums[i][j] = x;
			flag++; //上面两行为平移操作，平移结束后flag加一
			if (flag > k) flag = 1; //大于k则再次归一
		}
	}
	int sum = 0;
	for (j = 1; j <= n; j++) {
		for (i = 1; i <= n; i++) {
			sum += nums[i][j];
		}
		if (j == 1) printf("%d", sum);
		else printf(" %d", sum);
		sum = 0; //sum归零
	}
	return 0;
}