C++:: 프로그래머스 < 행렬의 곱셈 >

jahlee·2023년 7월 31일
0

프로그래머스_Lv.2

목록 보기
88/106
post-thumbnail

행렬의 곱셈을 구현하는 문제이다. 만들어 지는 벡터의 크기를 유의하며 풀면 크게 어렵지않은 문제이다.
ixj 행렬과 kxl 행렬을 곱하면 ixl크기의 행렬이 나온다.

#include <string>
#include <vector>
using namespace std;

vector<vector<int>> solution(vector<vector<int>> arr1, vector<vector<int>> arr2) {
    int row = arr1.size(), col = arr2[0].size();
    vector<vector<int>> answer(row, vector<int>(col, 0));
    
    for (int i=0; i<row; i++) {
        for (int j=0; j<col; j++) {
            for (int k=0; k<arr1[0].size(); k++) {
                answer[i][j] += arr1[i][k] * arr2[k][j];
            }
        }
    }
    return answer;
}

1개의 댓글

comment-user-thumbnail
2023년 7월 31일

많은 도움이 되었습니다, 감사합니다.

답글 달기