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

문제 URL

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

프로그래머스

위 문제는 2차원 행렬 arr1과 arr2를 입력받아, arr1에 arr2를 곱한 결과를 반환하는 문제이다.

 

문제를 푸는 방법은 첫번째 배열의 크기만큼 반복하면서 행렬의 곱셈에 맞게 두 배열의 위치를 조정하여 각 원소를 곱하고 곱한 값을 더하여 결과 행렬의 값을 만들어주면 되는 문제이다.

행렬의 곱

---

 

#include <string>
#include <vector>

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++){
        vector<int> temp;
        for(int j=0; j<arr2[0].size(); j++){
            int value = 0;
            for(int k=0; k<arr1[0].size(); k++)
                value += arr1[i][k] * arr2[k][j];
            temp.push_back(value);
        }
        answer.push_back(temp);
    }
    
    return answer;
}

참고: https://xtar.tistory.com/75