[数组]NC143 矩阵乘法-简单

​​NC143 矩阵乘法​​

描述

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

数据范围：，

要求：空间复杂度  ， 时间复杂度 

进阶：本题也有空间复杂度 ，时间复杂度 的解法

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

示例1

输入：

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

返回值：

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

复制

示例2

输入：

[[1]],[[1]]

返回值：

[[1]]

题解

#include <bits/stdc++.h>

using namespace std;

vector<vector<int>> solve(vector<vector<int>> &a, vector<vector<int>> &b)
{
  std::vector<std::vector<int>> res(a.size(), std::vector<int>(a.size(), 0));
  for (int i = 0; i < a.size(); ++i)
  {
    for (int k = 0; k < a.size(); ++k)
    {
      for (int x = 0; x < a.size(); ++x)
      {
        res[i][k] += a[i][x] * b[x][k];
      }
    }
  }
  return res;
}