[프로그래머스/Level 2] 행렬의 곱셈 (C++)

programmers.co.kr/learn/courses/30/lessons/12949

코딩테스트 연습 - 행렬의 곱셈

은근 헷갈렸던 문제

 

transferhwang.tistory.com/209

해당 문제의 심화(?) 문제이다.

 

#include <string>
#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++) { // arr1 행
        vector<int> tmp;
        for(int j = 0; j < arr2[0].size(); j++) { // arr2 행
            int sum = 0;
            for(int k = 0; k < arr1[0].size(); k++) { // arr1 열, arr2 행
                sum += arr1[i][k] * arr2[k][j];
            }
            tmp.push_back(sum);
        } 
        answer.push_back(tmp);
    }
    return answer;
}