알고리즘/프로그래머스 알고리즘

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

빠켱이 2021. 5. 5. 17:34
#include <string>
#include <vector>

using namespace std;

vector<vector<int>> solution(vector<vector<int>> arr1, vector<vector<int>> arr2) {
    vector<vector<int>> answer;
    int y = arr1.size();
    int x = arr1[0].size();
    int z = arr2[0].size();
    int tmp = 0, sum;
    for(int i  = 0; i < y; i++){
        vector <int> vec;
        for(int k = 0; k < z; k++){
            sum = 0;
            for(int j = 0; j < x; j++)
                sum += arr1[i][j] * arr2[j][k];
            vec.push_back(sum);
        }
        answer.push_back(vec);
    }
    return answer;
}