Python向量形式存储来处理矩阵相乘

原生设计思路:

矩阵相乘说到底就是行vector*列vector,在处理输入矩阵的时候就按照行vector和列vector存储,最后只需要两个for就可以。

时间复杂度O(n^2)

m, k, n = int(input()), int(input()), int(input())

mat1 = []
for _ in range(m):
    mat1.append(list(map(int, input().split())))
mat2 = [[] for _ in range(n)]
for _ in range(k):
    nums = list(map(int, input().split()))
    for j in range(n):
        mat2[j].append(nums[j])

for i in range(m):
    row = ''
    for j in range(n):
        temp = 0
        for a in range(k):
            temp += mat1[i][a]*mat2[j][a]
        row+=str(temp)+' '
    print(row.strip())

全部评论

相关推荐

不愿透露姓名的神秘牛友
06-27 14:11
很喜欢小米的新车,校招薪资每月22k,攒多久能买?
测试糕手手:别看工资,先看现金流存款。有50W存款以上再考虑,车是消耗品,选适合自己的重要。你有钱就当我没说过
点赞 评论 收藏
分享
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

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