pta 1097 矩陣行平移 （20 分）

1097 矩陣行平移 （20 分）

給定一個 n×n 的整數(shù)矩陣。對任一給定的正整數(shù) k

輸入格式：

輸入第一行給出 3 個正整數(shù)：n（<100）、k（

接下來 n 行，每行給出 n 個不超過 100 的正整數(shù)，為矩陣元素的值。數(shù)字間以空格分隔。

輸出格式：

在一行中輸出平移后第 1 到 n 列元素的和。數(shù)字間以 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 …… 這個規(guī)律順次向右平移），左端的空位用 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 
using namespace std;
int main() {
    int n,k,x;
    int ans[110],cnt=0;
    cin>>n>>k>>x;
    int p=0;
    int line[200][200];
    for(int i=0;i        if(i%2!=0){
            for(int j=0;j            cin>>line[i][j];
            }
        }else{
            p=p%k+1;
            for(int j=p;j                cin>>line[i][j];
            }
            for(int j=0;j        }
        
    }
    for(int i=0;i        int sum=0;
        for(int j=0;j            sum+=line[j][i];
        }
        ans[cnt++]=sum;
    }
    for(int i=0;i        if(i==0)cout<        else cout<<" "<    }
    return 0;
}