[ 프로그래머스 ] 행렬의 곱셈
문제
https://school.programmers.co.kr/learn/courses/30/lessons/12949
풀이
단순한 행렬곱 문제로, arr1의 행과, arr2의 열 크기의 2차원 벡터 안에 arr1의 행과 arr2의 열 간의 곱을 각각 곱한 후, 더해줘 값을 입력한다.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
#include <string>
#include <vector>
using namespace std;
vector<vector<int>> solution(vector<vector<int>> arr1, vector<vector<int>> arr2)
{
vector<vector<int>> answer(arr1.size(), vector <int>(arr2[0].size(), 0));
for (int i = 0; i < arr1.size(); i++)
{
for (int j = 0; j < arr2[0].size(); j++)
{
for (int k = 0; k < arr2.size(); k++)
{
answer[i][j] += (arr1[i][k]) * (arr2[k][j]);
}
}
}
return answer;
}
이 기사는 저작권자의 CC BY 4.0 라이센스를 따릅니다.