PAT 1097矩阵行平移(错题整理)

先给出原题目：

给定一个 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

我一开始的提交代码是：

#include<stdio.h>
int main()
{
    int n, k, x;
    int i, j, a, b, flag=1, temp;
    scanf("%d%d%d", &n, &k, &x);
    int num, sum, array[n][n];
    for (i=0; i<n; i++)
    {
        for (j=0; j<n; j++)
        {
            scanf("%d", &array[i][j]);
        }
        if ((i+1)%2 != 0)
        {
            if (flag)
            {
                for (a=n-1; a>0; a--)
                {
                    array[i][a] = array[i][a-1]; //数组中的所有元素后移;
                }
                array[i][0] = x;
                flag--;
            }
            else
            {
                for (a=n-1; a>=k; a--)
                {
                    array[i][a] = array[i][a-k];
                }
                for (b=0; b<k; b++)
                {
                    array[i][b] = x;
                }
                flag++;
            }
        }
    }
    for (i=0; i<n; i++)
    {
        sum = 0;
        for (j=0; j<n; j++)
        {
            sum += array[j][i];
        }
        if (i<n-1) printf("%d ", sum);
        else printf("%d", sum);

    }
    return 0;
}

        只得了12分，有几个测试点没有通过。这个算法思路是用了flag标志思想，对每一次平移进行标记，使得平移的规律为 1、2、1、2、1、2 …… ；判断出了什么时候要进行怎样的平移，然后再将数组的元素进行相应的平移，用了循环从数组的最后一个值开始向后平移，移到需要补空位的那个位置时就停止循环，然后再在循环外单独为这个空位进行赋值进行补空位，在平移k个位置时会采用两个循环，这样大大增加出错的可能性，果然在提交代码后只通过了两个测试点，大部分测试点没有通过。

借鉴大佬的方法：

        后来学习了其他大佬的写法，在移动元素的循环中就可补空位：

            int count=0, m=1;
            count=m;
            m++;
            if (m>k) m=1;
            for (a=n-1; a>=0; a--)
            {
                array[i][a] = array[i][a-count];
                if (a < count)
                {
                    array[i][a] = x;
                } 
            }

        在循环中当a=0时，其实数组是越界的，因为 a-1 = -1 ， 这时数组中补空位的地方是一个垃圾值，但是有了 之后的 a<count 判断使得，在补空位的元素上赋值了 x  。因此就在一个循环中做了两件事，一个是平移元素，一个是将需要补空位的地方赋上相应的值，这样大大降低了代码的复杂程度，非常值得学习。