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

 

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

프로그래머스

 

행렬의 곱셈 A x B

sum(A의 행 원소들 x B의 열 원소들)

 

예제)

   A           B

2 3 2    5 4 3 

4 2 4    2 4 1

3 1 4     3 1 1

 

1행 원소들만 해보자면

(2x5+3x2+2x3)

2 3 2    5 4 3

4 2 4    2 4 1

3 1 4     3 1 1

 

(2x4+3x4+2x1)

2 3 2    5 4 3

4 2 4    2 4 1

3 1 4     3 1 1

 

(2x3+3x1+2x1)

2 3 2    5 4 3

4 2 4    2 4 1

3 1 4     3 1 1

 

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