矩阵乘法

[编程题]矩阵乘法

热度指数：6554 时间限制：C/C++ 1秒，其他语言2秒空间限制：C/C++ 256M，其他语言512M
算法知识视频讲解

给定两个 n*n 的矩阵 A 和 B ，求 A*B 。

数据范围： $1 \le n \le 100$ ， $-100 \le Matrix_{i,j}\le 100$

要求：空间复杂度 $O(n^2)$ ，时间复杂度 $O(n^3 )$

进阶：本题也有空间复杂度 $O(n^2)$ ，时间复杂度 $O(n^{log7})$ 的解法

PS：更优时间复杂度的算法这里并不考察

示例1

输入

[[1,2],[3,2]],[[3,4],[2,1]]

输出

[[7,6],[13,14]]

示例2

输入

[[1]],[[1]]

输出

[[1]]

算法知识视频讲解

Python3

Oo啊哈

class Solution:
    def solve(self , a: List[List[int]], b: List[List[int]]) -> List[List[int]]:
        # write code here
        res = [[0]*len(b[0]) for i in range(len(a))]
        for i in range(len(a)):
            for j in range(len(b[0])):
                for k in range(len(b)):
                    res[i][j] += a[i][k]*b[k][j]
        return res

发表于 2022-04-09 20:09:04 回复(0)