输入整数m、p、n,再输入一个m行p列的整数矩阵A和一个p行n列的整数矩阵B,求两个矩阵的乘积AB
时间: 2024-04-30 13:18:31 浏览: 104
输入两个矩阵,计算两个矩阵相乘
以下是 Python 代码实现:
```python
m, p, n = map(int, input().split())
A = []
for i in range(m):
row = list(map(int, input().split()))
A.append(row)
B = []
for i in range(p):
row = list(map(int, input().split()))
B.append(row)
# 计算矩阵乘积
C = [[0] * n for _ in range(m)]
for i in range(m):
for j in range(n):
for k in range(p):
C[i][j] += A[i][k] * B[k][j]
# 输出结果
for i in range(m):
print(' '.join(str(x) for x in C[i]))
```
代码思路如下:
1. 首先读入三个整数 m、p、n;
2. 然后读入一个 m 行 p 列的整数矩阵 A 和一个 p 行 n 列的整数矩阵 B;
3. 声明一个 m 行 n 列的矩阵 C,用于存储乘积的结果;
4. 对于 C 的每个元素,都遍历 A 的对应行和 B 的对应列,计算乘积并累加;
5. 输出 C 中的所有元素。
注意,代码中使用了列表推导式和 join 函数,可以简化代码。
阅读全文