牛客题霸 矩阵乘法

题目描述

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

 

示例1

输入

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

返回值

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

 

备注:

n<=500 ，初始两个矩阵中每个数  0≤a_{ij}≤1000≤aij​≤100

 

Python代码：

class Solution:
    def solve(self , a , b ):
        # write code here
        n = len(a)
        mat = [[0 for _ in range(n)] for _ in range(n)]
        for i in range(n):
            for j in range(n):
                for k in range(n):
                    mat[i][j] += a[i][k] * b[k][j]
        return mat