[프로그래머스][Lv. 2] 행렬의 곱셈 C++ 풀이

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

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

 

풀이는 추후에 차차 올리도록 하겠습니다...

그냥 우리가 아는 행렬의 곱셈을 구현하면 됩니다.

 

 

 

 

#include <vector>
 
using namespace std;
 
int arr1row = -1, arr1col = -1, arr2col = -1;
int calMatrix(int i, int j, vector<vector<int>>& arr1, vector<vector<int>>& arr2) {
    int result = 0;
    for (int k = 0; k < arr1col; k++) {
        result += arr1[i][k] * arr2[k][j];
    }
    return result;
}
 
vector<vector<int>> solution(vector<vector<int>> arr1, vector<vector<int>> arr2) {
    vector<vector<int>> answer;
    arr1row = arr1.size(), arr1col = (*arr1.begin()).size(), arr2col = (*arr2.begin()).size();
        
    for (int i = 0; i < arr1row; i++) {
        answer.push_back(vector<int>());
        for (int j = 0; j < arr2col; j++) {
            answer[i].push_back(0);
        }
    }
    for (int i = 0; i < arr1row; i++) {
        for (int j = 0; j < arr2col; j++) {
            answer[i][j] = calMatrix(i, j, arr1, arr2);
        }
    }
    
    return answer;
}