알고리즘 · 코딩
[프로그래머스] 행렬의 곱셈
G A
2020. 3. 2. 20:29
문제 링크
https://programmers.co.kr/learn/courses/30/lessons/12949
코딩테스트 연습 - 행렬의 곱셈 | 프로그래머스
[[2, 3, 2], [4, 2, 4], [3, 1, 4]] [[5, 4, 3], [2, 4, 1], [3, 1, 1]] [[22, 22, 11], [36, 28, 18], [29, 20, 14]]
programmers.co.kr
C++ 풀이
#include <string>
#include <vector>
using namespace std;
vector<vector<int>> solution(vector<vector<int>> arr1, vector<vector<int>> arr2) {
vector<vector<int>> answer;
int arr2_Row_Size = arr2[0].size();
for (int i = 0; i < arr1.size(); i++) {
vector <int> x;
for (int k = 0; k < arr2_Row_Size; k++) {
int sum = 0;
for (int j = 0; j < arr1[i].size(); j++) {
sum += arr1[i][j] * arr2[j][k];
}
x.push_back(sum);
}
answer.push_back(x);
}
return answer;
}
행렬의 곱셈 과정을 그대로 적용시켜 풀었다.
반응형