【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

AC:

#include<bits/stdc++.h>
using namespace std;
const int N=110;
int a[N][N],b[N][N],c[N][N],d[N][N],e[N][N],sum[N];
int main()
{
    int n,k,x;
    cin>>n>>k>>x;
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
            cin>>a[i][j];
    int p=0,q=0;
    for(int i=0;i<n;i+=2)
    {
        for(int j=0;j<n;j++)
        {
            b[p][j]=a[i][j];
            d[q][j]=a[i+1][j];
        }
        p++;
        q++;
    }
    int m=0;
    for(int i=0;i<n;i++,m++)
    {
        if(m==k)
            m=0;
        for(int j=0;j<n;j++)
        {
            c[i][j+m+1]=b[i][j];
        }
    }
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<n;j++)
        {
         if(c[i][j]==0)
            c[i][j]=x;
        }
    }
    int cnt=0,t=0;
     for(int i=0;i<n/2+1;i++)
    {
        cnt++;
        for(int j=0;j<n;j++)
            e[t][j]=c[i][j];
        if(cnt==n/2+1)
            break;
            t++;
         for(int j=0;j<n;j++)
            e[t][j]=d[i][j];
        t++;
    }
      for(int j=0;j<n;j++)
    {
        for(int i=0;i<n;i++)
         sum[j]+=e[i][j];
    }
    for(int i=0;i<n;i++)
    {
        if(i==0) cout<<sum[i];
        else cout<<" "<<sum[i];
    }

}

猜你喜欢

转载自blog.csdn.net/m0_62504956/article/details/128660061