ccf-csp 2024-6-1 python

矩阵重塑（其一）

刷新 

时间限制： 1.0 秒

空间限制： 512 MiB

相关文件： 题目目录（样例文件）

题目背景

矩阵（二维）的重塑（reshape）操作是指改变矩阵的行数和列数，同时保持矩阵中元素的总数不变。

题目描述

矩阵的重塑操作可以具体定义为以下步骤：

设原矩阵为 MM，其维度为 n×mn×m，即有 nn 行和 mm 列。新矩阵为 M′M′，其维度为 p×qp×q。重塑操作要满足 n×m=p×qn×m=p×q，这保证了元素的总数不变。

1. 线性化原矩阵：按照行优先的顺序，将原矩阵 MM 的元素转换成一个长度为 n×mn×m 的一维数组 AA。这意味着你先读取 MM 的第 00 行元素，然后是第 11 行，依此类推，直到最后一行。

2. 填充新矩阵：使用一维数组 AA 中的元素按照行优先的顺序填充新矩阵 M′M′。首先填充 M′M′ 的第 00 行，直到该行有 qq 个元素，然后继续填充第 11 行，直到所有 pp 行都被填满。

给定原矩阵中的一个元素的位置 (i,j)(i,j)（0≤i<n0≤i<n 且 0≤j<m0≤j<m），我们可以找到这个元素在被线性化后的一维数组 AA 中的位置 kk（0≤k<n×m0≤k<n×m），然后确定它在新矩阵 M′M′ 中的位置 (i′,j′)(i′,j′)（0≤i′<p0≤i′<p 且 0≤j<q0≤j<q）。它们之间满足如下数学关系：i×m+j=k=i′×q+j′i×m+j=k=i′×q+j′

给定 n×mn×m 的矩阵 MM 和目标形状 pp、qq，试将 MM 重塑为 p×qp×q 的矩阵 M′M′。

输入格式

从标准输入读入数据。

输入共 n+1n+1 行。

输入的第一行包含四个正整数 nn、mm 和 pp、qq。

接下来依次输入原矩阵 MM 的第 00 到第 n−1n−1 行，每行包含 mm 个整数，按列下标从 00 到 m−1m−1 的顺序依次给出。

输出格式

输出到标准输出。

输出共 pp 行，每行 qq 个整数，表示重塑后的矩阵 M′M′。输出格式与输入相同，即依次输出 M′M′ 的第 00 行到第 p−1p−1 行；行内按列下标从 00 到 q−1q−1 的顺序输出，且两个整数间仅用一个空格分隔。

样例1输入

2 3 3 2
1 2 3
4 5 6

 

样例1输出

1 2
3 4
5 6

 

样例2输入

2 2 1 4
6 6
6 6

 

样例2输出

6 6 6 6

 

子任务

全部的测试数据满足：

• nn、mm 和 pp、qq 均为正整数且 n×m=p×q≤104n×m=p×q≤104；

• 输入矩阵中每个元素的绝对值不超过 10001000。

提示

评测环境仅提供各语言的标准库，特别地，不提供任何线性代数库（如 numpy、pytorch 等）。

 

 

——————————————————————————

 

 

        分析：就按站题目给的定义一步一步实现下来就行，暴力遍历，不需要优化。

        具体思路细节见代码。

n, m ,p, q = map(int, input().split())

# 输入原矩阵M
M =[]
for _ in range(n):
    M.append( list(map( int , input().split()) ) )

# 转化为一维矩阵A
A = []
for i in range(n):
    for j in range(m):
        A.append( M[i][j])

# 转化为新矩阵M_1
i =0
M_1 = []
while( i<= (len(A) - q)):
    j = i+ q-1
    M_1.append(list( A[i:j+1] ))
    i += q

# 输出M_1矩阵
for i in range(p):
    for j in range(q):
        print(M_1[i][j],end = " ")
    print("")