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

在安全路上的白白 lv.1

发布时间：2022-11-26 13:30:09 341

描述

给定两个 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;
}

文章来源： https://blog.51cto.com/u_6650004/5508862

特别声明：以上内容（图片及文字）均为互联网收集或者用户上传发布，本站仅提供信息存储服务！如有侵权或有涉及法律问题请联系我们。