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

全部评论

相关推荐

开心的豹子最喜欢夏天:看公司项目,有的公司是自己做完了之后卖给实体产业使用,一种是定制化软件,这种就是外包
点赞 评论 收藏
分享
10-24 19:32
西北大学 Java
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务