[프로그래머스] 행렬의 곱셈

---

🔗 문제 링크

https://school.programmers.co.kr/learn/courses/30/lessons/12949

프로그래머스

---

👩‍💻 코드

#include <vector>

using namespace std;

vector<vector<int>> solution(vector<vector<int>> arr1, vector<vector<int>> arr2) {
    int m = arr1.size();
    int n = arr1[0].size();
    int p = arr2[0].size();
    
    vector<vector<int>> answer(m, vector<int>(p, 0));
    
    for (int i = 0; i < m; i++) {
        for (int j = 0; j < p; j++) {
            for (int k = 0; k < n; k++) {
                answer[i][j] += arr1[i][k] * arr2[k][j];
            }
        }
    }
    
    return answer;
}

---

📝 풀이

$$ answer_{ij}  = arr1_{i0}arr2_{0i} + arr1_{i1}arr2_{1i} + ... + arr1_{in}arr2_{ni} $$

answer의 성분은 arr1의 i번째 행과 arr2의 j번째 열의 성분들을 각각 더해 곱한 것이다.

수식에서 0부터 n까지 증가하는 변수가 코드에서 k이다.

행렬의 모든 원소를 채우고 answer을 return한다.