C++ Algorithm & Study/C++ & Algorithm Strategies
[Programmers] 행렬의 곱셈
GameChoi
2023. 1. 23. 19:15
https://school.programmers.co.kr/learn/courses/30/lessons/12949
프로그래머스
코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.
programmers.co.kr
0. Headers
#include <vector>
using namespace std;
1. 알고리즘
- 행렬의 곱셈은 배열1의 첫번 째 가로줄 * 배열2의 첫번째 세로줄을 곱하는 것임
- 따라서 위의 식을 곱하고 배열에 넣으면 됨
vector<vector<int>> solution(vector<vector<int>> arr1, vector<vector<int>> arr2) {
vector<vector<int>> answer;
for (int i = 0; i < arr1.size(); i++)
{
vector<int> v;
for (int j = 0; j < arr2[0].size(); j++)
{
int temp = 0;
for (int k = 0; k < arr1[0].size(); k++)
temp += arr1[i][k] * arr2[k][j];
v.push_back(temp);
}
answer.push_back(v);
}
return answer;
}
2. 완성 코드
#include <vector>
using namespace std;
vector<vector<int>> solution(vector<vector<int>> arr1, vector<vector<int>> arr2) {
vector<vector<int>> answer;
for (int i = 0; i < arr1.size(); i++)
{
vector<int> v;
for (int j = 0; j < arr2[0].size(); j++)
{
int temp = 0;
for (int k = 0; k < arr1[0].size(); k++)
temp += arr1[i][k] * arr2[k][j];
v.push_back(temp);
}
answer.push_back(v);
}
return answer;
}