행렬의 곱셈을 구현하는 문제이다. 만들어 지는 벡터의 크기를 유의하며 풀면 크게 어렵지않은 문제이다.
ixj 행렬과 kxl 행렬을 곱하면 ixl크기의 행렬이 나온다.
#include <string>
#include <vector>
using namespace std;
vector<vector<int>> solution(vector<vector<int>> arr1, vector<vector<int>> arr2) {
int row = arr1.size(), col = arr2[0].size();
vector<vector<int>> answer(row, vector<int>(col, 0));
for (int i=0; i<row; i++) {
for (int j=0; j<col; j++) {
for (int k=0; k<arr1[0].size(); k++) {
answer[i][j] += arr1[i][k] * arr2[k][j];
}
}
}
return answer;
}
많은 도움이 되었습니다, 감사합니다.