矩阵乘法

[编程题]矩阵乘法

热度指数：6307 时间限制：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]]

算法知识视频讲解

JavaScript

别卷啦！

function solve( a ,  b ) {
    // write code here
    let m = a.length
    let matrix = Array.from(new Array(m), ()=> new Array(m).fill(0)) 
    for(let i = 0 ;i<m; i++){
        for(let j = 0;j<m; j++){
            for(let k =0;k<m; k++ ){
               matrix[i][j] += a[i][k]* b[k][j]  
            }
        }
    }
    return matrix
}

发表于 2021-01-12 12:22:56 回复(0)