题解 | #矩阵乘法#

矩阵乘法

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

import sys

# 获取a,b数组的行/列数量
a_row = int(input())
a_column = b_row = int(input())
b_column = int(input())

# 根据2个矩阵行数,获取输入组成a\b数组
a_arr = [list(map(int, input().split())) for i in range(int(a_row))]
b_arr = [list(map(int, input().split())) for i in range(int(b_row))]
# print(a_arr, b_arr)

# 初始化乘积详情数组
product_arr = list()

# a数组的行数做主循环, b数组的列数做次循环
for i in range(a_row):
    for j in range(b_column):
        # k来自b_row循环,即满足公式A(ik) * B(kj)
        summary = sum([a_arr[i][k]*b_arr[k][j] for k in range(b_row)])
        product_arr.append(str(summary))

# a_row做循环,b_column作为数组分割间隔
for k in range(1, a_row+1):
    print(" ".join(product_arr[(k-1)*b_column:(k)*b_column]))

全部评论

相关推荐

点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务