L1-8 矩阵A乘以B-Python题解-CSDN博客

本文链接：https://blog.csdn.net/m0_73616862/article/details/137650802

本文介绍了一个计算矩阵乘法的编程任务，强调了矩阵A的列数必须等于矩阵B的行数这一条件，并提供了示例输入和预期输出的格式说明。

摘要由CSDN通过智能技术生成

给定两个矩阵A和B，要求你计算它们的乘积矩阵AB。需要注意的是，只有规模匹配的矩阵才可以相乘。即若A有Ra行、Ca列，B有Rb行、Cb列，则只有Ca与Rb相等时，两个矩阵才能相乘。

输入格式：

输入先后给出两个矩阵A和B。对于每个矩阵，首先在一行中给出其行数R和列数C，随后R行，每行给出C个整数，以1个空格分隔，且行首尾没有多余的空格。输入保证两个矩阵的R和C都是正数，并且所有整数的绝对值不超过100。

输出格式：

若输入的两个矩阵的规模是匹配的，则按照输入的格式输出乘积矩阵AB，否则输出Error: Ca != Rb，其中Ca是A的列数，Rb是B的行数。

输入样例1：

2 3
1 2 3
4 5 6
3 4
7 8 9 0
-1 -2 -3 -4
5 6 7 8

输出样例1：

2 4
20 22 24 16
53 58 63 28

输入样例2：

输出样例2：

Error: 2 != 3

代码长度限制 16 KB

时间限制 400 ms

内存限制 64 MB

解答：

r1, c1 = map(int, input().split())
a = [[0] * 110 for _ in range(110)]
b = [[0] * 110 for _ in range(110)]
for i in range(r1):
    a[i] = list(map(int, input().split()))
r2, c2 = map(int, input().split())
for i in range(r2):
    b[i] = list(map(int, input().split()))
if c1 == r2:
    print(r1, c2)
    for i in range(r1):
        for j in range(c2):
            sum1 = 0
            for k in range(c1):
                sum1 += a[i][k] * b[k][j]
            if j == c2-1:
                print(sum1)
            else :
                print(sum1,end=" ")
else:
    print("Error: %d != %d" % (c1, r2))

逻辑非常简单，值得一提的是：

#初始化两个列表
a = [[0] * 110 for _ in range(110)]
b = [[0] * 110 for _ in range(110)]

a = [[0] * 110] * 110
b = [[0] * 110] * 110

这两种列表初始化方式是不同的，简单的说，上者a，b是两个不同列表，操作a，b不会相互影响，而下者a，b实际上是一个列表。我们可以看看chatgpt是怎么说的。

2.输出矩阵的时候每一行后面不能有空格，所以不能简单的写：

print(sum1,end=" ")

而是要判断有没有到最后一个数字，否则就会格式错误。