[PTA]矩阵列平移

2023-12-20 03:40:45

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

输出样例:

440 399 369 421 302 386 428

样例解读

需要平移的是第 2、4、6 列。给定?k=2,应该将这三列顺次整体向下平移 1、2、1 位(如果有更多列,就应该按照 1、2、1、2 …… 这个规律顺次向下平移),顶端的空位用 99 来填充。平移后的矩阵变成:

11 99 23 99 20 99 89
37 87 27 99 63 75 11
44 94 50 67 40 50 24
73 38 63 91 62 26 68
15 85 27 26 88 18 43
23 83 98 28 30 25 99
77 78 48 97 25 81 22

?解题思路:

首先可以得到每一列应该移动的数目,可以创建一个长度为n的数组来实现,也可以在用的时候(列为偶数的时候)计算(这里使用数组实现)具体计算方法如下:

  • 定义一个flag = 0;
  • 如果i为偶数,则在第i列平移的次数为flag%k+1;flag++;
  • 遍历count数组,如果是奇数列(i%2==0),置为0;否则为flag%k+1;flag++;

接着,创建一个vector容器v,以列的方式遍历二维数组,将每一列的数据存入v中,在v的头部插入count[i]个x;再重新设置v的大小为n;(此时v中的数据是该列的数据,不能直接求和输出)

最后,创建一个res数组,长度为n,初始化为0;res[i]表示第i行的和,不断将每一个v中的数据累加到res中,得到最后的结果.

代码实现:

#include <iostream>
#include <vector>
using namespace std;

/**
 * 矩阵列平移(PTA)
 * @return
 */
int main(){
    int n = 0,k = 0,x = 0;
    cin>>n>>k>>x;
    int arr[n][n];
    for (int i = 0; i < n; ++i) {
        for (int j = 0; j < n; ++j) {
            cin>>arr[i][j];
        }
    }

    int count[n];//计算每一列需要移动的次数,按照测试样例中count数组为:0 1 0 2 0 1 0
    int res[n];//用于计算每一行的求和结果
    int flag = 0;//使用flag%k+1来得到每一个count[i]的值
    for (int i = 0; i < n; ++i) {
        res[i] = 0;
        if(i%2==0){
            //奇数列
            count[i] = 0;
        }else{
            count[i] = flag%k+1;
            flag++;
        }
    }
    vector<int> v;
    for (int i = 0; i < n; ++i) {
        //先将一列的数据放到vector容器中,尾插
        for (int j = 0; j < n; ++j) {
            v.push_back(arr[j][i]);
        }
        //在头部插入count[i]个x;如果count[i]==0,就插入0个
        v.insert(v.begin(),count[i],x);
        //重新设置容器的大小为n
        v.resize(n);
        //使用swap来收缩容器
        vector<int>(v).swap(v);
        for (int j = 0; j < n; ++j) {
            res[j]+=v[j];
        }
        //清空容器
        v.clear();
    }
    for (int i = 0; i < n; ++i) {
        i==n-1?cout<<res[i]:cout<<res[i]<<" ";
    }
    return 0;
}

文章来源:https://blog.csdn.net/m0_67428547/article/details/135091211
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。