행렬의 곱셈

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

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

 

#include <string>
#include <vector>

using namespace std;

vector<vector<int>> solution(vector<vector<int>> arr1, vector<vector<int>> arr2) {
    vector<vector<int>> answer;
    int n = arr1.size();
    int m = arr2[0].size();
    vector<vector<int>>v(n, vector<int>(m,0)); 
    for(int i=0; i<n; i++){
        for(int j=0; j<m; j++){
            int sum=0;
            for(int p=0; p<arr1[i].size(); p++){
                sum+=arr1[i][p]*arr2[p][j];
            }
            v[i][j]=sum;
        }
    }
    
    answer = v;
    
    return answer;
}

 

- 특정 크기의 이차원 배열 선언하는 법 잊지 말기 :vector<vector<int>>v(n, vector<int>(m,0));