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())

全部评论

相关推荐

不愿透露姓名的神秘牛友
07-09 13:05
TMD找工作本来就烦,这东西什么素质啊😡
Beeee0927:hr是超雄了,不过也是有道理的
点赞 评论 收藏
分享
07-10 12:17
已编辑
商丘师范学院 Java
后来123321:别着急,我学院本大二,投了1100份,两个面试,其中一个还是我去线下招聘会投的简历,有时候这东西也得看运气
无实习如何秋招上岸
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

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