프로그래머스 행렬의 곱셈 - C++

문제 설명

내풀이

행렬곱 공식은 AXB 행렬과 BXC 행렬을 곱하면 AXC행렬이 되며, B가 같을 때만 곱할 수 있음. 이것만 알고있다면 다른 문제와 달리 긴 풀이 없이 간단히 풀 수 있는 문제였습니다.

코드

#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 sum = 0;
            for(int k=0; k<arr1[0].size(); k++)
            {
                sum += arr1[i][k] * arr2[k][j];
            }
            v.push_back(sum);
        }
        answer.push_back(v);
    }

    return answer;
}