题解 | 矩阵乘法

import sys

x,y,z = input(),input(),input()
a1 = []
a2 = []
x1 = 0
for i in sys.stdin:
    x1 += 1
    a = list(map(int,i.split()))
    if x1 > int(x):
        a2.append(a)
    else:
        a1.append(a)
        #print(a1)
count = []
for i in range(int(x)):  
    for k in range(int(z)):
        adds = 0
        #k = 0 
        for j in range(int(y)):
            adds += a1[i][j] * a2[j][k]
        #print(adds)
        count.append(adds)
i =  0
for k in range(int(x)):
    s = ''  
    for l in range(int(z)):
        if l == int(z) - 1:
            s += str(count[i])
        else:
            s += str(count[i]) + " "
        i += 1
        #print(i)
    print(s)