题解 | #矩阵乘法#

矩阵乘法

https://www.nowcoder.com/practice/ebe941260f8c4210aa8c17e99cbc663b

根据输入得到A,B两个矩阵,然后直接根据题目中的公式求输出矩阵。

import sys

x, y, z = int(input()), int(input()), int(input())
A = [[0] * y for _ in range(x)]
B = [[0] * z for _ in range(y)]
ans = [[0] * z for _ in range(x)]
for i in range(x):
    A[i] =[int(m) for m in input().split(' ')]
for i in range(y):
    B[i] = [int(m) for m in input().split(' ')]
for i in range(x):
    for j in range(z):
        cal = 0
        for k in range(y):
            cal += A[i][k] * B[k][j]
        ans[i][j] = cal
for i in range(x):
    out = [str(m) for m in ans[i]]
    print(' '.join(out))




全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务